Я пытаюсь запустить модель Наивного Байеса, используя MultinomialNB, а затем предсказать настроение рецензий на фильмы.Я запускаю следующий код:
from sklearn.feature_extraction.text import CountVectorizer
from nltk import RegexpTokenizer
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import MultinomialNB
movie_revs = pd.read_csv("film.tsv", sep='\t')
token = RegexpTokenizer(r'[a-zA-Z0-9]+')
vectorizer = CountVectorizer(lowercase=True, stop_words='english', ngram_range = (1,1), tokenizer = token.tokenize)
movie_revs_counts = vectorizer.fit_transform(movie_revs.Phrase)
movie_revs_x = movie_revs['Phrase']
movie_revs_y = movie_revs['Sentiment']
x_train, x_test, y_train, y_test = train_test_split(movie_revs_x, movie_revs_y, test_size = 0.3, random_state = 4)
y_train = y_train.astype('int')
x_traincv = vectorizer.fit_transform(x_train)
x_testcv = vectorizer.transform(x_test)
mnb.fit(x_traincv, y_train)
В этот момент я сталкиваюсь с ошибкой:
"TypeError: fit() missing 1 required positional argument: 'y'"
Но я не вижу, в чем проблема.Любая помощь приветствуется.