0
私はMultinomialNBの作業をしようとしています。私はトレーニングとテストセットでCountVectorizerを使用していますが、もちろん両方のsetzには異なる単語があります。だから私は、なぜエラーが表示されるCountVectorizer MultinomialNB ValueError:ディメンションの不一致
ValueError: dimension mismatch
が発生しますが、私はそれを修正する方法はわかりません。私はCountVectorizerは右どのように使用できるか
NotFittedError: CountVectorizer - Vocabulary wasn't fitted.
他のポスト(SciPy and scikit-learn - ValueError: Dimension mismatch)で提案されたようCountVectorizer().transform
の代わりCountVectorizer().fit_transform
を試みたが、それはちょうど私
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.cross_validation import train_test_split
from sklearn.naive_bayes import MultinomialNB
from sklearn.metrics import classification_report
import sklearn.feature_extraction
df = data
y = df["meal_parent_category"]
X = df['name_cleaned']
X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.3)
X_train = CountVectorizer().fit_transform(X_train)
X_test = CountVectorizer().fit_transform(X_test)
algo = MultinomialNB()
algo.fit(X_train,y_train)
y = algo.predict(X_test)
print(classification_report(y_test,y_pred))
あなたが '' X_test = vectorizer_train.transform(X_test)を使用している可能性が、新しいものを定義する代わりに。 –
ああ、私はそれを湖。ありがとうございました :) –