model = RandomForestClassifier(random_state=1, n_estimators=10, min_samples_split=2, min_samples_leaf=1) model.fit(train_titanic, train_label) predictions = model.predict(df_test) result = pd.DataFrame({'PassengerId':titanic_test['PassengerId'].as_matrix(), 'Survived':predictions.astype(np.int32)}) result.to_csv("random_forest_predictions.csv", index=False) print(pd.read_csv("random_forest_predictions.csv"))
时间: 2024-03-30 20:33:29 浏览: 29
RandomForest_sklearn.zip_sklearn_sklearn RF_southern9qq_随机森林
这段代码是使用随机森林算法对 Titanic 数据集进行分类,并将预测结果保存到 CSV 文件中。其中,模型的参数设置为:随机种子为 1,决策树数量为 10,最小分割样本数为 2,最小叶子节点样本数为 1。训练数据集为 train_titanic,标签为 train_label,测试数据集为 df_test。最后将预测结果以 PassengerId 和 Survived 两列的形式保存到 CSV 文件中,并打印出来。
阅读全文