我正在学习机器学习课程,并且已按照说明学习了决策树的简单示例。该代码有效,直到我能预测结果为止。我收到以下错误:
““ ValueError:预期的2D数组,取而代之的是1D数组:array = [10。1. 4. 0. 0. 0.]。如果数据具有单个功能,则可以使用array.reshape(-1,1)重塑数据。或array.reshape(1,-1)如果包含单个样本。”
代码:clf = RandomForestClassifier(n_estimators = 10)clf = clf.fit(X,y)
print(clf.predict([10, 1, 4, 0, 0, 0]))
print(clf.predict([10, 0, 4, 0, 0, 0]))
我不确定如何“重塑”预测电话。我尝试使用谷歌搜索,但对我来说没有意义。
只需在您的预测函数中传递[[10,1,4,0,0,0]]而不是[10,1,4,0,0,0],则通过传递1D数组就意味着您正在传递6个具有单个功能,现在通过将其设为2D,这意味着您要传递一个具有6个功能的示例,并对其进行了模型训练
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句