search

massquantity / LibRecommender

Versatile End-to-End Recommender System
https://librecommender.readthedocs.io/
MIT License
366 stars 65 forks source link

New Model #454

Open Masoumeh-Alinia opened 6 months ago

Masoumeh-Alinia commented 6 months ago

Hello, first of all thanks for your great work with the library! I had a question. I want to use the library's defined model from LightGCN and add another series of layers and then train the model. It is possible? How can I do this because the input for the next layer must be tensor.

massquantity commented 6 months ago

Hi, you can inherit from the LightGCN class. Internally it uses a pytorch module self.torch_model to perform the forward propagation. So you can add layers after this module.