Skip to content

Instantly share code, notes, and snippets.

View EthanRosenthal's full-sized avatar

Ethan Rosenthal EthanRosenthal

340 followers · 16 following
View GitHub Profile
@EthanRosenthal
EthanRosenthal / ExplicitMF.py
Created March 13, 2016 20:54
Class for training explicit matrix factorization model using either ALS or SGD
class ExplicitMF():
def __init__(self,
ratings,
n_factors=40,
learning='sgd',
item_fact_reg=0.0,
user_fact_reg=0.0,
item_bias_reg=0.0,
user_bias_reg=0.0,
verbose=False):
@EthanRosenthal
EthanRosenthal / ExplicitMF_ALS.py
Created March 13, 2016 20:38
Class for ALS training of an explicit matrix factorization model
from numpy.linalg import solve
class ExplicitMF():
def __init__(self,
ratings,
n_factors=40,
item_reg=0.0,
user_reg=0.0,
verbose=False):
"""