Final results for 2022 machine learning term project work results
result : recommendation system
Content-based Filtering | Collaborative Filtering |
---|---|
K-means , Cosine similarity | EM , Matrix Factorization |
Using Two Dataset for recommendation system
- https://www.kaggle.com/datasets/sengzhaotoo/movielens-small : Collaborative Filtering
- https://www.kaggle.com/datasets/rounakbanik/the-movies-dataset : Content-based Filtering
Each movie is classified by cluster and a recommendation system is created based on the keywords of movies in each cluster.
- Code : https://github.com/m4a1carbin4/2022-ML_Term-project/blob/main/Content-based-Filtering(K-means).ipynb
Create a rating table between user and movies. After clustering them, a recommendation system is created by predicting user ratings for each cluster.
- Code : https://github.com/m4a1carbin4/2022-ML_Term-project/blob/main/Collaborative Filtering(EM).ipynb
2022-ML_Term-project/Collaborative Filtering(EM).ipynb at main · m4a1carbin4/2022-ML_Term-project