在上一篇 用逻辑回归实现鸢尾花的分类(1)中,我们了解了鸢尾花数据集中的特征数据等信息,并尝试使用Logistic Regression方法基于scikit提供的iris数据集做简单的分类。这篇进阶版会带大家来学习如何将原始文件中的数据转变成机器学习算法可用的numpy数据。相信这对于无论是入门数据分析者还是有一定数据分析基础的数据分析师,都是在实际操作处理现实原始数据时最犯难的一件事。巧妇难为无米之炊,没有能够进行机器学习的数据,纵然你有各种机器学习算法的能力与技能,还是一样不能做好一个数据分析项目。
所以这次会带大家来了解一下基于平衡的样本下一个大体完整的数据处理与分析过程。此外,这篇notebook会在模型构造时运用sklearn中的一个有意思的类:Pipeline,即管道机制,来实现流式教程的封装与管理(streaming workflows with pipelines)。
来自官网对于Pipeline的解读: pipeline of transforms with a final estimator. 管道的中间步骤都是转换‘Transforms’,即实现fit与transform方法,最终的estimator只需要实现fit。旨在于组装几个步骤从而可以在设置不同参数的时候做交叉验证。
In [1]:
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import plotly.plotly as py
import plotly.graph_objs as go
from plotly.offline import init_notebook_mode, iplot
init_notebook_mode(connected = True)
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler
from sklearn.decomposition import PCA
from sklearn.pipeline import Pipeline
from sklearn.linear_model import LogisticRegression
from sklearn import metrics
In [2]:
iris_path = '/home/kesci/input/iris/iris.csv'
iris = pd.read_csv(iris_path)
In [3]:
iris.head()
Out[3]:
In [4]:
# S1:
# 映射函数iris_type: 将string的label映射至数字label
# s: 品种的名字
def iris_type(s):
class_label = {'setosa':0, 'versicolor':1, 'virginica':2}
return class_label[s]
In [5]:
# S2: 将第4列内容映射至iris_type函数定义的内容
pd.io.parsers.read_csv(iris_path, converters = {4:iris_type}).head()
Out[5]:
In [6]:
# S3: 将上面转变过且解析至dataframe的
data = np.array(pd.io.parsers.read_csv(iris_path, converters = {4:iris_type}))
data[:10,:] # 查看前10行的数据
Out[6]:
In [7]:
# Step 4:将原始数据集划分成训练集与测试集
# 用np.split按列(axis=1)进行分割
# (4,):分割位置,前4列作为x的数据,第4列之后都是y的数据
x,y = np.split(data, (4,), axis = 1)
X = x[:,0:2] # 取前两列特征
# 用train_test_split将数据按照7:3的比例分割训练集与测试集,
# 随机种子设为1(每次得到一样的随机数),设为0或不设(每次随机数都不同)
x_train, x_test, y_train,y_test = train_test_split(x,y,test_size = 0.3,random_state = 0)
2. 模型的搭建与训练
- Pipeline(steps) 利用sklearn提供的管道机制Pipeline来实现对全部步骤的流式化封装与管理。
- 第一个环节:可以先进行 数据标准化 StandardScaler()
- 中间环节:可以加上 PCA降维处理 取2个重要特征
- 最终环节:逻辑回归分类器
In [8]:
pipe_LR = Pipeline([
# ('sc', StandardScaler()),
# ('pca', PCA(n_components = 2)),
('clf', LogisticRegression(random_state=1))
])
# 开始训练
pipe_LR.fit(x_train, y_train)
Out[8]:
In [9]:
print("训练集准确率: %0.3f" %pipe_LR.score(x_train, y_train))
In [10]:
print("测试集准确率: %0.3f" %pipe_LR.score(x_test, y_test))
In [11]:
y_hat = pipe_LR.predict(x_test)
accuracy = metrics.accuracy_score(y_test, y_hat)
print("逻辑回归分类器的准确率:%0.3f" % accuracy)
In [12]:
target_names = ['setosa', 'versicolor', 'virginica']
print(metrics.classification_report(y_test, y_hat, target_names = target_names))