This is a simpla recommender system to predict the movie rating.
The implemnted algorithms are nearest neighbors
, slope one
and hybrid algorithm
.
PS: slope one
has a poor performance, I don't come up with a good idea to determine the similarity between movie.
- Python3
-
Download and extract ml-1m.zip in the repo folder
-
pip3 install -r requirements.txt
-
python3 conver_to_sqlite3.py
-
python3 recommender.py
-
analysis.py
is a simple script to check the RMSD for each algorithm. -
result.txt
is a sample output
When seed = 0, sample number = 100, RMSD is
############################# Average Rating #############################
1.0099504938362078
########################### Nearest Neighbour ############################
0.565685424949238
############################### Slope One ################################
1.224744871391589
############################ Hybrid Algorithm ############################
0.5099019513592785