Using Multinomial Naive Bayes, this code snippet illustrates a text classification task. It starts by loading training and testing data sets containing text descriptions and the associated genres.
machine-learning
text-classification
machine-learning-algorithms
word-embeddings
movie-database
naive-bayes-classifier
support-vector-machine
svm-classifier
nltk-library
multinomial-naive-bayes
tfidf-vectorizer
movie-genre-classification
movie-genre-prediction
dataset-movie-genre
-
Updated
Apr 8, 2024 - Jupyter Notebook