值错误:传递的项目数量错误2,位置意味着1

时间:2017-10-03 09:26:13

标签: pandas machine-learning valueerror

此代码工作正常......

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

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']])