beta-team / beta-recsys

Beta-RecSys: Build, Evaluate and Tune Automated Recommender Systems
https://beta-recsys.readthedocs.io/en/latest/
MIT License
162 stars 33 forks source link

Fixed LightGCN error #426

Closed JavierSanzCruza closed 2 years ago

JavierSanzCruza commented 2 years ago

There was an error on LightGCN which caused the dropout function to execute in evaluation time. I added a call to self.eval(), so training is equal to false each time the predict function is called.