0
我試圖運行MultinomiaL Naive bayes並收到以下錯誤。給出樣本訓練數據。測試數據完全相似。ValueError:發現具有不一致樣本數的輸入變量:[1,14048]
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
你的'text_train'和'targets_train'的形狀是什麼?另外我不認爲MultinomialNB會默認處理文本數據。你需要預處理它。 –