У попередній вправі ви згенерували bag-of-words подання для тренувальних і тестових рецензій на фільми. У цій вправі ми використаємо цю модель, щоб натренувати класифікатор Наївного Байєса, який може визначати сентимент рецензії, і обчислити його точність. Зауважте, що оскільки це задача бінарної класифікації, модель здатна відносити рецензію лише до позитивних (1) або негативних (0). Вона не вміє виявляти нейтральні рецензії.

Якщо ви не пам'ятаєте, тренувальні й тестові BoW-вектори доступні як X_train_bow і X_test_bow відповідно. Відповідні мітки доступні як y_train і y_test відповідно. Крім того, для довідки, початковий набір даних рецензій на фільми доступний як df.

Побудова класифікатора Naive Bayes на основі BoW

Create Your Free Account