ktm icon indicating copy to clipboard operation
ktm copied to clipboard

How can I save a trained model?

Open laizef opened this issue 5 years ago • 3 comments

Hello, I found that the model was trained and used for prediction in a single function FM.run(), but how can I save or load a trained model? I would appreciate it if the problem could be addressed.

laizef avatar Oct 20 '19 11:10 laizef

I got it... The model can not be saved when we choose the MCMC method.

laizef avatar Oct 21 '19 05:10 laizef

Actually you can get the model weights! See on this Mangaki project.

jilljenn avatar Oct 22 '19 12:10 jilljenn

I will add some functions to do the same on KTM. On maybe it would be better to do it directly on pywFM.

jilljenn avatar Oct 22 '19 12:10 jilljenn

It is now done: https://github.com/jilljenn/ktm/blob/master/fm.py#L26 Please try and let me know.

jilljenn avatar Dec 27 '23 15:12 jilljenn