我是ML和sklearn的新手。我试图在一个数据集上使用GaussianNB,X_train[2500,800]
,Y_train[2500,8]
。
from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()
clf.fit(X, Y)
运行程序时,它显示
ValueError: bad input shape (2500, 8).
如何将Y_train[2500,8]
转换为Y_train[2500,1]
?
Tags:
OP使用的是一个单热编码器,因此fit函数不能与数组一起工作@Ishant Mrinal建议这样做
这将允许您将一个热编码传递到fit函数中。
相关问题 更多 >
编程相关推荐