我是ML和sklearn领域的新手。我尝试在具有X_train[2500,800]，Y_train[2500,8]的数据集上使用GaussianNB。
from sklearn.naive_bayes import GaussianNB
clf = GaussianNB()
clf.fit(X, Y)
在运行程序时，它显示
ValueError：输入形状错误（2500，8）。
如何将
Y_train[2500,8]转换为Y_train[2500,1]？ 最佳答案
OP使用的是一个热编码器，因此拟合函数不适用于数组@Ishant Mrinal建议使用此功能
Y_train = np.argmax(Y_train, axis=1)
这样您就可以将一种热门编码传递给fit函数。