我正在尝试运行MultinomiaL朴素贝叶斯并收到以下错误。提供样本培训数据。测试数据完全相似。
def main():
text_train, targets_train = read_data('train')
text_test, targets_test = read_data('test')
classifier1 = MultinomialNB()
classifier1.fit(text_train, targets_train)
prediction1 = classifier1.predict(text_test)
样本数据:
Train:
category, text
Family, I love you Mom
University, I hate this course
有时候我会面对这个问题并从错误中找出大部分原因是输入数据应该是二维数组,例如如果要构建回归模型。你写这个代码,然后你将面临这个错误!
例如:
a = np.array([1,2,3]).T
b = np.array([4,5,6]).T
regr = linear_model.LinearRegression()
regr.fit(a, b)
那你应该加点东西!
a = np.array([[1,2,3]]).T
b = np.array([[4,5,6]]).T
最后你会正常运行!所以这只是我的经验!这只是一个参考,而不是一个标准答案!我来自中国学习英语和蟒蛇的学生!