Projekt för Pythonkurs (AI)
| README.md | ||
Supervised Learning - Movie/TV-Show recommender
Specification
Movie/TV-Show recommender
This program will recommend you what movie or th-show to view based on what Movie/TV-Show you like.
Data Source:
I will use 4 datasets from kaggle, 3 datasets from streaming-sites Netflix, Amazon Prime and Disney Plus, also 1 from a IMDB dataset.
Model:
I will use k-Nearest Neighbors (k-NN) alhorithm that can help me find other titles based on features like Title, Release year, Description, Cast, Director and genres.
Features:
- Load data and preprocessing before creating new dataset csv file.
- Model training with k-NN algorithm.
Requirements:
- Title data:
- Title
- Genres
- Release year
- Cast
- Director
- Description
- User data:
- What Movie / TV-Show
- What genre
- Director
Libraries
- pandas: Data manipulation and analysis
- scikit-learn: machine learning algorithms and preprocessing
- numpy: numerical operations
- beatifulsoup4: web scraping
Classes
- LoadData
- Loading, cleaning and saving alla data to csv
- check_data
- clean_text
- clean_data
- load_dataset
- create_data
- save_data
- load_data
- UserData
- input
- Recommendations
- get_recommendations