Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

How to deploy model using node js #669

Open
s-bh opened this issue Nov 22, 2022 · 1 comment
Open

How to deploy model using node js #669

s-bh opened this issue Nov 22, 2022 · 1 comment

Comments

@s-bh
Copy link

s-bh commented Nov 22, 2022

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?

@np-n
Copy link

np-n commented Jul 4, 2023

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!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants