I föregående övning skapade du bag-of-words-representationer för tränings- och testdata med filmrecensioner. I den här övningen använder vi den modellen för att träna en Naïve Bayes-klassificerare som kan avgöra sentimentet i en filmrecension och beräkna dess noggrannhet. Eftersom det här är ett binärt klassificeringsproblem kan modellen bara klassificera en recension som antingen positiv (1) eller negativ (0) – den kan inte hantera neutrala recensioner.

Om du behöver en påminnelse: tränings- och test-BoW-vektorerna finns tillgängliga som X_train_bow respektive X_test_bow. Motsvarande etiketter finns som y_train respektive y_test. Den ursprungliga datamängden med filmrecensioner finns tillgänglig som df.

Bygga en naiv Bayes-klassificerare med BoW

Create Your Free Account