当前位置:   article > 正文

机器学习——使用pipeline简化预处理,构建模型过程_pipeline导入模型

pipeline导入模型

一:使用pipeline进行数据预处理,模型构建

from sklearn.pipeline import Pipeline
from sklearn.svm import SVC
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import MinMaxScaler

# load and split the data
cancer = load_breast_cancer()
X_train, X_test, y_train, y_test = train_test_split(cancer.data, cancer.target, random_state=0)

pipe = Pipeline([("scaler", MinMaxScaler()), ("svm", SVC())])#有两个步骤

pipe.fit(X_train, y_train) #自动对数据进行scaler,再用scale之后的数据输入SVC构建模型
print("Test score: {:.2f}".format(pipe.score(X_test, y_test)))
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14

在这里插入图片描述
二:将pipeline应用到GridSearchCV

#接着上面的代码
param_grid = {'svm__C': [0.001, 0.01, 0.1, 1, 10, 100],
'svm__gamma': [0.001, 0.01, 0.1, 1, 10, 100]}
#注意这里参数svm__C:
  • 1
  • 2
  • 3
声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号