matrix-factorization is a light-weight program written in python language for performing basic operations for matrix factorization-based collaborative filtering. I have plans to create a python module from this repository in the future. If you want to contribute to this project, you are most welcome.
download the matrix-factorization.py file and put it in the same folder as your main.py. After that follow the code snippet below.
import matrix-factorization as mf
#matrix_factorization(data-> numpy array,features-> integer,user_features-> numpy_array (optional), item_features-> numpy array (optional) )
#creating object
d = matrix_factorization(data,features)
#train_model(learning_rate-> can be anything.(Default= 0.01) , iteration-> can be anything (Default=1000))
d.train_model(0.1,10000)
print(d.predicted_matrix())
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.
Please make sure to update tests as appropriate.