-
Data Collection and Preprocessing: Gathered and cleaned a dataset of movies, including metadata like titles, descriptions, and genres, using sources like IMDb or TMDb. Applied natural language processing (NLP) techniques to preprocess text data (e.g., tokenization, removing stop words).
-
Feature Extraction: Extracted key features from movie descriptions and other metadata using techniques like TF-IDF, word embeddings , transformers to create numerical representations for model input.
-
Model Development and Training: Built and trained classification models using machine learning techniques like Naive Bayes, Random Forests to predict movie genres based on the text description.
-
Evaluation and Optimization: Evaluated model performance using metrics like accuracy, precision, recall, and F1-score. Fine-tuned hyperparameters to improve classification performance and handled multi-label classification challenges where movies have multiple genres.
-
Deployment and Results: Documented results, highlighting model accuracy and improvements over baseline models of genre overlap in multi-label classification.
-
Notifications
You must be signed in to change notification settings - Fork 0
Krishnasoni200320/Movie-Class-Genre-Classification
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Repository files navigation
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published