此代码工作正常......
forest1 = RandomForestClassifier()
forest1.fit(train[['Random Forest Score','lr','neural']],train['target'])
但是当我试着预测时,
test['target'] = forest1.predict_proba(test[['Random Forest Score','lr','neural']])
显示错误..
ValueError:传递的项目数量错误2,展示位置意味着1
答案 0 :(得分:1)
forest1.predict_proba(...)预测X 的等级概率。
返回shape = [n_samples, n_classes]
的数组或n_outputs列表
如果n_outputs>这样的数组1.输入样本的类概率。类的顺序对应于属性类_。
test['target']
需要一个向量(1D数组)
尝试使用predict()
代替predict_proba
:
test['target'] = forest1.predict(test[['Random Forest', 'Score','lr','neural']])