lightfm icon indicating copy to clipboard operation
lightfm copied to clipboard

How to deploy model using node js

Open s-bh opened this issue 1 year ago • 1 comments

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) in ----> 1 model.save("model.h5")

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?

s-bh avatar Nov 22 '22 18:11 s-bh

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!

np-n avatar Jul 04 '23 06:07 np-n