Skip to content

bebound/recommender

Repository files navigation

A simple Recommender System

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.

Requirments

  • Python3

Usage

  • 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

Screenshot

screenshot

Sample Result

When seed = 0, sample number = 100, RMSD is

#############################    Average Rating    #############################
1.0099504938362078
###########################    Nearest Neighbour    ############################
0.565685424949238
###############################    Slope One    ################################
1.224744871391589
############################    Hybrid Algorithm    ############################
0.5099019513592785

About

A simple CF recommender

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages