将数据划分为训练集和测试集；缩放特征区间

导入葡萄酒数据：

import numpy as np
import pandas as pd

df_wine = pd.read_csv("https://archive.ics.uci.edu/ml/machine-learning-databases/wine/wine.data", header=None)
df_wine.columns = ["class label", "alcohol",
                   "malic acid", "ash",
                   "alcalinity of ash", "magnesium",
                   "total phenols", "flavanoids",
                   "nonflavaniod phenols", "proanthocyanins",
                   "color intensity", "hue",
                   "OD280/OD315 of diluted wines", "proline"]
# 查看类标
print("class label:", np.unique(df_wine["class label"]))
print(df_wine.head())

运行结果：

 

划分训练集和测试集：

 　　我们可以使用 sklearn.model_selection 中的 train_test_split 划分数据，test_size用来设置测试数据的比例，random_state用来

设置随机数是否保持一致。

from sklearn.model_selection import train_test_split
# import warnings
# warnings.filterwarnings('ignore')
x, y = df_wine.iloc[:, 1:].values, df_wine.iloc[:, 0].values
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.3, random_state=0)

　　这里如果你用的是 sklearn.cross_validation 的 train_test_split ,那么代码是会报警告的，由于版本的更新，推荐使用上面的代码。

 

特征缩放：

　　特征缩放我们可以采用归一化和标准化两者方法

# 特征缩放：归一化
from sklearn.preprocessing import MinMaxScaler
mms = MinMaxScaler()
x_train_norm = mms.fit_transform(x_train)
x_test_norm = mms.transform(x_test)
print(x_test_norm, "\n")

# 特征缩放：标准化
from sklearn.preprocessing import StandardScaler
stdsc = StandardScaler()
x_train_std = stdsc.fit_transform(x_train)
x_test_std = stdsc.transform(x_test)
print(x_test_std)