值错误：通过的项目数错误 2，位置暗示 1

Question

这段代码运行良好...

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

Answer 1

forest1.predict_proba(...) 预测类别X 的概率。

它返回一个 shape = [n_samples, n_classes] 的数组，或者一个 n_outputs 的列表 如果 n_outputs > 1，则为此类数组。输入样本的类概率。类的顺序与属性classes_中的顺序相对应。

test['target'] 需要一个向量（一维数组）

尝试使用predict() 而不是predict_proba：

test['target'] = forest1.predict(test[['Random Forest', 'Score','lr','neural']])