lightfm
lightfm copied to clipboard
How to deploy model using node js
I created a recommendation system using lightFM on Google Colab. I want to deploy this model using Node JS. I read this article and tried the method recommended here. It recommended using tensorflow.js. For that the model needed to be saved and this is where the problem arose since I wrote my model using lightFM.
This gave rise to following error:
`
AttributeError Traceback (most recent call last)
AttributeError: 'LightFM' object has no attribute 'save' `
What is the alternative to model.save in lightfm? How can we deploy model written in lightfm using node js?
You can save the model using the serialization library called pickle
and load the model too.
import pickle
from lightfm import LightFM
# instantiate model
# train your LightFM model here
# Save the trained model to a pickle file
with open("./model_file.pkl", "wb") as file:
pickle.dump(model, file)
and load the model:
import pickle
from lightfm import LightFM
# read the pickle file
with open("./model_file.pkl", "rb") as file:
model = pickle.load(file)
I hope this will help to load and save the model. Thank you!