I'm trying to create a collaborative filtering algorithm to suggest products to certain users.
I started shortly and started working with TensorFlow (I thought it was sufficiently effective and flexible). I found this code that does what I'm interested in, creates the model and train the user IDs, products, and ratings: https://github.com/songgc/TF-recomm
I launched the code and trained the model.
After training the model I would need to make the predictions, that is, get suggestions for each user so that they can be saved in a DB from which I access with a NODE.js application.
How do I retrieve this list of suggestions for each user when the training is done?
if __name__ == '__main__':
df_train, df_test=get_data()
svd(df_train, df_test)
print("Done!")