vot = VotingClassifier(estimators=[('lr', lr),('xgb',xgb_model), ('rf', rf),('gbdt',gbdt)], voting='hard')
vot.fit(train,y_train)
y_pred = vot.predict(test)
# print(y_pred)
submit['y'] = y_pred
submit.to_csv('my_voting_prediction.csv',index=False)
1个回答
你是在做回归还是分类？
如果是分类的话，你需要把y的数值类型转换成int
如果是回归的话，VotingClassifier不支持回归，因为它是对分类器结果进行投票
SofaSofa数据科学社区DS面试题库 DS面经
matt
2018-09-22 11:06
做的是回归，知道了，谢谢
-
evan
2018-09-22 14:14
相关讨论
随便看看