pyFM
pyFM copied to clipboard
About regularization
Hi,
From the code I see:
# Regularization Parameters (start with no regularization)
self.reg_0 = 0.0
self.reg_w = 0.0
self.reg_v = np.repeat(0.0, num_factors)
However, I don't see where the regularization parameters are updated. Moreover, after fitting the model, I used model.reg_v to output the regularization parameters, it gave me an array of zeros. I am wondering does the model impose regularization on the model parameters? Thanks
Hi,
I add arguments and activate the regularizations.
PULL REQUEST: https://github.com/coreylynch/pyFM/pull/42 MY FORK BRANCH: https://github.com/uchida-takumi/pyFM/tree/available_regularization%2330
IF your help, I am happy.