我正在尝试使用sklearn中的MultinomialNB来对一些数据进行分类。我创建了一个包含一些标记的训练数据的样本csv文件,我希望用它来训练模型,但收到以下错误信息:
ValueError: Expected 2D array, got 1D array instead: array=[0 1 2 2].
我知道这是一个非常小的数据集,但是一旦代码工作起来,我会添加更多的数据。
这是我的代码:
回答:
尝试这样做:
x_train = x_train.values.reshape(-1, 1)y_train = y_train.values.reshape(-1, 1)
numpy
的reshape操作不是就地进行的。所以你传递给分类器的数组实际上是旧的形状。