深度學(xué)習(xí) Kaggle playground模型融合練習(xí)賽(ML·二分類任務(wù))
X_test = dict_vec.fit_transform(X_test.to_dict(orient='records'))rfc = RandomForestClassifier()cross_val_score(rfc, X_train, Y_train, cv=5).mean()# 執(zhí)行訓(xùn)練及預(yù)測,并將預(yù)測結(jié)果寫入文件,注意要和比賽要求的文件格式一致。rfc.fit(X_train, Y_train)rfc_y_predict = rfc.predict(X_test)rfc_submission = pd.DataFrame({'PassengerId': test['PassengerId'], 'Survived': rfc_y_predict})rfc_submission.to_csv('dataset/rfc_submission.csv')
標(biāo)簽: