ktm
ktm copied to clipboard
How can I save a trained model?
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.
I got it... The model can not be saved when we choose the MCMC method.
Actually you can get the model weights! See on this Mangaki project.
I will add some functions to do the same on KTM. On maybe it would be better to do it directly on pywFM.
It is now done: https://github.com/jilljenn/ktm/blob/master/fm.py#L26 Please try and let me know.