【发布时间】:2020-12-29 16:17:59
【问题描述】:
目前,我可以通过使用make_column_transformer 和make_pipeline 来构建一个看起来像这样的模型:
from sklearn.compose import make_column_transformer
from sklearn.model_selection import cross_val_score
from sklearn.preprocessing import OneHotEncoder
from sklearn.tree import DecisionTreeClassifier
from sklearn.pipeline import make_pipeline
[in]: dtc = DecisionTreeClassifier()
[in]: column_trans = make_column_transformer(
(OneHotEncoder(handle_unknown='ignore'), ['var1', 'var2',
'var3', 'var4', 'var5', 'var6'
]),
remainder='passthrough')
[in]: column_trans.fit_transform(X)
[in] pipe = make_pipeline(column_trans, dtc)
[in]: cross_val_score(pipe, X_train, y_train, cv=5, scoring='accuracy').mean()
[out]: ... prediction
我浏览了文档,似乎找不到任何可以将流程简化为 gridsearchCV 的内容
【问题讨论】:
标签: python-3.x pandas machine-learning scikit-learn gridsearchcv