Large scale matrix factorization on GPU. Instructions to run:
import MF
mf = MF(size_of_square_matrix, dim_of_factors, regularization_coef, learning_rate)
mf.mat_fact(num_iters)
W, H = mf.get_factors()
Requirements:
- Theano
- Numpy
-
Notifications
You must be signed in to change notification settings - Fork 2
shashankg7/Matrix-Factorization-GPU
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Large scale matrix factorization on GPU
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published