jilljenn/ktm

Making predictions

Closed this issue · 2 comments

clabra commented

Hi Jill-Jênn,

Are you considering add a predict function to the code to get predictions once the model is trained?. Thanks for sharing this project

Yes should have done it for a while

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