【发布时间】:2017-05-24 10:21:51
【问题描述】:
我对在测试数据集中使用 cross cross_val_predict 感到困惑。
我创建了一个简单的随机森林模型并使用 cross_val_predict 进行预测
from sklearn.ensemble import RandomForestClassifier
from sklearn.cross_validation import cross_val_predict, KFold
lr = RandomForestClassifier(random_state=1, class_weight="balanced", n_estimators=25, max_depth=6)
kf = KFold(train_df.shape[0], random_state=1)
predictions = cross_val_predict(lr,train_df[features_columns], train_df["target"], cv=kf)
predictions = pd.Series(predictions)
我对下一步感到困惑,我如何使用上面学到的知识对测试数据集进行预测?
【问题讨论】:
-
你必须先
fit你的模型,然后你可以打电话给predict。
标签: python machine-learning scikit-learn data-science