翻译
光城责编
郭芮
scikit-learn提供最先进的机器学习算法。但是,这些算法不能直接用于原始数据。原始数据需要事先进行预处理。因此,除了机器学习算法之外,scikit-learn还提供了一套预处理方法。此外,scikit-learn提供用于流水线化这些估计器的连接器(即变压器,回归器,分类器,聚类器等)。
在本文中,将介绍scikit-learn功能集,允许流水线估计器、评估这些流水线、使用超参数优化调整这些流水线以及创建复杂的预处理步骤。
基本用例:训练和测试分类器
对于第一个示例,我们将在数据集上训练和测试一个分类器。我们将使用此示例来回忆scikit-learn的API。
我们将使用digits数据集,这是一个手写数字的数据集。
#完成数据集的加载fromsklearn.datasetsimportload_digits#return_X_y默认为False,这种情况下则为一个Bunch对象,改为True,可以直接得到(data,target)X,y=load_digits(return_X_y=True)
X中的每行包含64个图像像素的强度,对于X中的每个样本,我们得到表示所写数字对应的y。
#下面完成灰度图的绘制#灰度显示图像plt.imshow(X[0].reshape(8,8),cmap=gray);#关闭坐标轴plt.axis(off)#格式化打印print(Thedigitintheimageis{}.format(y[0]))
输出:Thedigitintheimageis0
在机器学习中,我们应该通过在不同的数据集上进行训练和测试来评估我们的模型。train_test_split是一个用于将数据拆分为两个独立数据集的效用函数,stratify参数可强制将训练和测试数据集的类分布与整个数据集的类分布相同。
#划分数据为训练集与测试集,添加stratify参数,以使得训练和测试数据集的类分布与整个数据集的类分布相同。fromsklearn.model_selectionimporttrain_test_splitX_train,X_test,y_train,y_test=train_test_split(X,y,stratify=y,random_state=42)
一旦我们拥有独立的培训和测试集,我们就可以使用fit方法学习机器学习模型。我们将使用score方法来测试此方法,依赖于默认的准确度指标。
#求出Logistic回归的精确度得分fromsklearn.linear_modelimportLogisticRegressionclf=LogisticRegression(solver=lbfgs,multi_class=auto,max_iter=,random_state=42)clf.fit(X_train,y_train)accuracy=clf.score(X_test,y_test)print(Accuracyscoreofthe{}is{:.2f}.format(clf.__class__.__name__,accuracy))AccuracyscoreoftheLogisticRegressionis0.95
scikit-learn的API在分类器中是一致的。因此,我们可以通过RandomForestClassifier轻松替换LogisticRegression分类器。这些更改很小,仅与分类器实例的创建有关。
#RandomForestClassifier轻松替换LogisticRegression分类器fromsklearn.ensembleimportRandomForestClassifierclf=RandomForestClassifier(n_estimators=,n_jobs=-1,random_state=42)clf.fit(X_train,y_train)accuracy=clf.score(X_test,y_test)print(Accuracyscoreofthe{}is{:.2f}.format(clf.__class__.__name__,accuracy))
输出:
AccuracyscoreoftheRandomForestClassifieris0.96
练习
完成接下来的练习:
加载乳腺癌数据集,从sklearn.datasets导入函数load_breast_cancer:#%loadsolutions/01_1_solutions.py使用sklearn.model_selection.train_test_split拆分数据集并保留30%的数据集以进行测试。确保对数据进行分层(即使用stratify参数)并将random_state设置为0:#%loadsolutions/01_2_solutions.py使用训练数据训练监督分类器:#%loadsolutions/01_3_solutions.py使用拟合分类器预测测试集的分类标签:#%loadsolutions/01_4_solutions.py计算测试集的balanced精度,需要从sklearn.metrics导入balanced_accuracy_score:#%loadsolutions/01_5_solutions.py
更高级的用例:在训练和测试分类器之前预处理数据
2.1标准化数据
在学习模型之前可能需要预处理。例如,一个用户可能对创建手工制作的特征或者算法感兴趣,那么他可能会对数据进行一些先验假设。在我们的例子中,LogisticRegression使用的求解器期望数据被规范化。因此,我们需要在训练模型之前标准化数据。为了观察这个必要条件,我们将检查训练模型所需的迭代次数。
fromsklearn.linear_modelimportLogisticRegressionclf=LogisticRegression(solver=lbfgs,multi_class=auto,max_iter=,random_state=42)clf.fit(X_train,y_train)print({}required{}iterationstobefitted.format(clf.__class__.__name__,clf.n_iter_[0]))
输出:
LogisticRegressionrequirediterationstobefitted
MinMaxScaler变换器用于规范化数据。该标量应该以下列方式应用:学习(即,fit方法)训练集上的统计数据并标准化(即,transform方法)训练集和测试集。最后,我们将训练和测试这个模型并得到归一化后的数据集。
fromsklearn.preprocessingimportMinMaxScalerfromsklearn.linear_modelimportLogisticRegressionscaler=MinMaxScaler()X_train_scaled=scaler.fit_transform(X_train)X_test_scaled=scaler.transform(X_test)clf=LogisticRegression(solver=lbfgs,multi_class=auto,max_iter=0,random_state=42)clf.fit(X_train_scaled,y_train)accuracy=clf.score(X_test_scaled,y_test)print(Accuracyscoreofthe{}is{:.2f}.format(clf.__class__.__name__,accuracy))print({}required{}iterationstobefitted.format(clf.__class__.__name__,clf.n_iter_[0]))
输出:
AccuracyscoreoftheLogisticRegressionis0.96LogisticRegressionrequirediterationstobefitted
通过归一化数据,模型的收敛速度要比未归一化的数据快得多(迭代次数变少了)。
2.2错误的预处理模式
我们强调了如何预处理和充分训练机器学习模型。发现预处理数据的错误方法也很有趣。其中有两个潜在的错误,易于犯错但又很容易发现。
第一种模式是在整个数据集分成训练和测试集之前标准化数据。
scaler=MinMaxScaler()X_scaled=scaler.fit_transform(X)X_train_prescaled,X_test_prescaled,y_train_prescaled,y_test_prescaled=train_test_split(X_scaled,y,stratify=y,random_state=42)clf=LogisticRegression(solver=lbfgs,multi_class=auto,max_iter=0,random_state=42)clf.fit(X_train_prescaled,y_train_prescaled)accuracy=clf.score(X_test_prescaled,y_test_prescaled)print(Accuracyscoreofthe{}is{:.2f}.format(clf.__class__.__name__,accuracy))
输出:
AccuracyscoreoftheLogisticRegressionis0.96
第二种模式是独立地标准化训练和测试集。它回来在训练和测试集上调用fit方法。因此,训练和测试集的标准化不同。
scaler=MinMaxScaler()X_train_prescaled=scaler.fit_transform(X_train)#这里发生了变化(将transform替换为fit_transform)X_test_prescaled=scaler.fit_transform(X_test)clf=LogisticRegression(solver=lbfgs,multi_class=auto,max_iter=0,random_state=42)clf.fit(X_train_prescaled,y_train)accuracy=clf.score(X_test_prescaled,y_test)print(Accuracyscoreofthe{}is{:.2f}.format(clf.__class__.__name__,accuracy))
输出:
AccuracyscoreoftheLogisticRegressionis0.96
2.3保持简单,愚蠢:使用scikit-learn的管道连接器
前面提到的两个模式是数据泄漏的问题。然而,当必须手动进行预处理时,很难防止这种错误。因此,scikit-learn引入了Pipeline对象。它依次连接多个变压器和分类器(或回归器)。我们可以创建一个如下管道:
fromsklearn.pipelineimportPipelinepipe=Pipeline(steps=[(scaler,MinMaxScaler()),(clf,LogisticRegression(solver=lbfgs,multi_class=auto,random_state=42))])
我们看到这个管道包含了缩放器(归一化)和分类器的参数。有时,为管道中的每个估计器命名可能会很繁琐,而make_pipeline将自动为每个估计器命名,这是类名的小写。
fromsklearn.pipelineimportmake_pipelinepipe=make_pipeline(MinMaxScaler(),LogisticRegression(solver=lbfgs,multi_class=auto,random_state=42,max_iter=0))
管道将具有相同的API。我们使用fit来训练分类器和socre来检查准确性。然而,调用fit会调用管道中所有变换器的fit_transform方法。调用score(或predict和predict_proba)将调用管道中所有变换器的内部变换。它对应于本文2.1中的规范化过程。
pipe.fit(X_train,y_train)accuracy=pipe.score(X_test,y_test)print(Accuracyscoreofthe{}is{:.2f}.format(pipe.__class__.__name__,accuracy))AccuracyscoreofthePipelineis0.96
我们可以使用get_params()检查管道的所有参数。
pipe.get_params()
输出:
{logisticregression:LogisticRegression(C=1.0,class_weight=None,dual=False,fit_intercept=True,intercept_scaling=1,max_iter=0,multi_class=auto,n_jobs=None,penalty=l2,random_state=42,solver=lbfgs,tol=0.,verbose=0,warm_start=False),logisticregression__C:1.0,.........}
练习
重用第一个练习的乳腺癌数据集来训练,可以从linear_model导入SGDClassifier。使用此分类器和从sklearn.preprocessing导入的StandardScaler变换器来创建管道,然后训练和测试这条管道。
#%loadsolutions/02_solutions.py
当更多优于更少时:交叉验证而不是单独拆分
分割数据对于评估统计模型性能是必要的。但是,它减少了可用于学习模型的样本数量。因此,应尽可能使用交叉验证。有多个拆分也会提供有关模型稳定性的信息。
scikit-learn提供了三个函数:cross_val_score,cross_val_predict和cross_validate。后者提供了有关拟合时间,训练和测试分数的更多信息。我也可以一次返回多个分数。
fromsklearn.model_selectionimportcross_validatepipe=make_pipeline(MinMaxScaler(),LogisticRegression(solver=lbfgs,multi_class=auto,max_iter=0,random_state=42))scores=cross_validate(pipe,X,y,cv=3,return_train_score=True)
使用交叉验证函数,我们可以快速检查训练和测试分数,并使用pandas快速绘图。
importpandasaspddf_scores=pd.DataFrame(scores)df_scores
输出:
#pandas绘制箱体图df_scores[[train_score,test_score]].boxplot()
输出:
练习
使用上一个练习的管道并进行交叉验证,而不是单个拆分评估。
#%loadsolutions/03_solutions.py
超参数优化:微调管道内部
有时希望找到管道组件的参数,从而获得最佳精度。我们已经看到我们可以使用get_params()检查管道的参数。
pipe.get_params()
输出:
{logisticregression:LogisticRegression(C=1.0,class_weight=None,dual=False,fit_intercept=True,intercept_scaling=1,max_iter=0,multi_class=auto,n_jobs=None,penalty=l2,random_state=42,solver=lbfgs,tol=0.,verbose=0,warm_start=False),logisticregression__C:1.0,.........}
可以通过穷举搜索来优化超参数。GridSearchCV提供此类实用程序,并通过参数网格进行交叉验证的网格搜索。
如下例子,我们希望优化LogisticRegression分类器的C和penalty参数。
fromsklearn.model_selectionimportGridSearchCVpipe=make_pipeline(MinMaxScaler(),LogisticRegression(solver=saga,multi_class=auto,random_state=42,max_iter=))param_grid={logisticregression__C:[0.1,1.0,10],logisticregression__penalty:[l2,l1]}grid=GridSearchCV(pipe,param_grid=param_grid,cv=3,n_jobs=-1,return_train_score=True)grid.fit(X_train,y_train)
输出:
GridSearchCV(cv=3,error_score=raise-deprecating,.........scoring=None,verbose=0)
在拟合网格搜索对象时,它会在训练集上找到最佳的参数组合(使用交叉验证)。我们可以通过访问属性cv_results_来得到网格搜索的结果。通过这个属性允许我们可以检查参数对模型性能的影响。
df_grid=pd.DataFrame(grid.cv_results_)df_grid
输出:
默认情况下,网格搜索对象也表现为估计器。一旦它被fit后,调用score将超参数固定为找到的最佳参数。
grid.best_params_
输出:
{logisticregression__C:10,logisticregression__penalty:l2}
此外,可以将网格搜索称为任何其他分类器以进行预测。
accuracy=grid.score(X_test,y_test)print(Accuracyscoreofthe{}is{:.2f}.format(grid.__class__.__name__,accuracy))AccuracyscoreoftheGridSearchCVis0.96
最重要的是,我们只对单个分割进行网格搜索。但是,如前所述,我们可能有兴趣进行外部交叉验证,以估计模型的性能和不同的数据样本,并检查性能的潜在变化。由于网格搜索是一个估计器,我们可以直接在cross_validate函数中使用它。
scores=cross_validate(grid,X,y,cv=3,n_jobs=-1,return_train_score=True)df_scores=pd.DataFrame(scores)df_scores
输出:
练习
重复使用乳腺癌数据集的先前管道并进行网格搜索以评估hinge(铰链)andlog(对数)损失之间的差异。此外,微调penalty。
#%loadsolutions/04_solutions.py
总结:我的scikit-learn管道只有不到10行代码(跳过import语句)
importpandasaspdfromsklearn.preprocessingimportMinMaxScalerfromsklearn.linear_modelimportLogisticRegressionfromsklearn.pipelineimportmake_pipelinefromsklearn.model_selectionimportGridSearchCVfromsklearn.model_selectionimportcross_validatepipe=make_pipeline(MinMaxScaler(),LogisticRegression(solver=saga,multi_class=auto,random_state=42,max_iter=))param_grid={logisticregression__C:[0.1,1.0,10],logisticregression__penalty:[l2,l1]}grid=GridSearchCV(pipe,param_grid=param_grid,cv=3,n_jobs=-1)scores=pd.DataFrame(cross_validate(grid,X,y,cv=3,n_jobs=-1,return_train_score=True))scores[[train_score,test_score]].boxplot()
输出:
异构数据:当您使用数字以外的数据时
到目前为止,我们使用scikit-learn来训练使用数值数据的模型。
X
输出:
array([[0.,0.,5.,...,0.,0.,0.],[0.,0.,0.,...,10.,0.,0.],[0.,0.,0.,...,16.,9.,0.],...,[0.,0.,1.,...,6.,0.,0.],[0.,0.,2.,...,12.,0.,0.],[0.,0.,10.,...,12.,1.,0.]])
X是仅包含浮点值的NumPy数组。但是,数据集可以包含混合类型。
importosdata=pd.read_csv(os.path.join(data,titanic_openml.csv),na_values=?)data.head()
输出:
泰坦尼克号数据集包含分类、文本和数字特征。我们将使用此数据集来预测乘客是否在泰坦尼克号中幸存下来。
让我们将数据拆分为训练和测试集,并将幸存列用作目标。
y=data[survived]X=data.drop(columns=survived)X_train,X_test,y_train,y_test=train_test_split(X,y,random_state=42)
首先,可以尝试使用LogisticRegression分类器,看看它的表现有多好。
clf=LogisticRegression()clf.fit(X_train,y_train)
哎呀,大多数分类器都设计用于处理数值数据。因此,我们需要将分类数据转换为数字特征。最简单的方法是使用OneHotEncoder对每个分类特征进行读热编码。让我们以sex与embarked列为例。请注意,我们还会遇到一些缺失的数据。我们将使用SimpleImputer用常量值替换缺失值。
fromsklearn.imputeimportSimpleImputerfromsklearn.preprocessingimportOneHotEncoderohe=make_pipeline(SimpleImputer(strategy=constant),OneHotEncoder())X_encoded=ohe.fit_transform(X_train[[sex,embarked]])X_encoded.toarray()
输出:
array([[0.,1.,0.,0.,1.,0.],[0.,1.,1.,0.,0.,0.],[0.,1.,0.,0.,1.,0.],...,[0.,1.,0.,0.,1.,0.],[1.,0.,0.,0.,1.,0.],[1.,0.,0.,0.,1.,0.]])
这样,可以对分类特征进行编码。但是,我们也希望标准化数字特征。因此,我们需要将原始数据分成2个子组并应用不同的预处理:(i)分类数据的独热编;(ii)数值数据的标准缩放(归一化)。我们还需要处理两种情况下的缺失值:对于分类列,我们将字符串missing_values替换为缺失值,该字符串将自行解释为类别。对于数值数据,我们将用感兴趣的特征的平均值替换缺失的数据。
分类数据的独热编:
col_cat=[sex,embarked]col_num=[age,sibsp,parch,fare]X_train_cat=X_train[col_cat]X_train_num=X_train[col_num]X_test_cat=X_test[col_cat]X_test_num=X_test[col_num]
数值数据的标准缩放(归一化):
fromsklearn.preprocessingimportStandardScalerscaler_cat=make_pipeline(SimpleImputer(strategy=constant),OneHotEncoder())X_train_cat_enc=scaler_cat.fit_transform(X_train_cat)X_test_cat_enc=scaler_cat.transform(X_test_cat)scaler_num=make_pipeline(SimpleImputer(strategy=mean),StandardScaler())X_train_num_scaled=scaler_num.fit_transform(X_train_num)X_test_num_scaled=scaler_num.transform(X_test_num)
我们应该像在本文2.1中那样在训练和测试集上应用这些变换。
importnumpyasnpfromscipyimportsparseX_train_scaled=sparse.hstack((X_train_cat_enc,sparse.csr_matrix(X_train_num_scaled)))X_test_scaled=sparse.hstack((X_test_cat_enc,sparse.csr_matrix(X_test_num_scaled)))
转换完成后,我们现在可以组合所有数值的信息。最后,我们使用LogisticRegression分类器作为模型。
clf=LogisticRegression(solver=lbfgs)clf.fit(X_train_scaled,y_train)accuracy=clf.score(X_test_scaled,y_test)print(Accuracyscoreofthe{}is{:.2f}.format(clf.__class__.__name__,accuracy))
输出:
AccuracyscoreoftheLogisticRegressionis0.79
上面首先转换数据然后拟合/评分分类器的模式恰好是本节2.1的模式之一。因此,我们希望为此目的使用管道。
但是,我们还希望对矩阵的不同列进行不同的处理。应使用ColumnTransformer转换器或make_column_transformer函数。它用于在不同的列上自动应用不同的管道。
fromsklearn.