search

massquantity / LibRecommender

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

About when using DatasetPure.build_trainset #497

Open kyorisann opened 3 weeks ago

kyorisann commented 3 weeks ago

HI! I am trying to use the data from movie rens. The algorithm is NFC_neural_collaborative_filtering. When "DatasetPure.build_trainset" is used to split the data during training, Is the data of user and item encoded? If so, when looking at the original data of user and item after the prediction, is it correct to map the original ID from "data_info.user2id"? Is this correct?

massquantity commented 3 weeks ago

Yes, DatasetPure.build_trainset will encode the data. The predict function will convert the original user and item internally, so you do not need to map it by yourself.