search

SSSxCCC / Recommender-System

A developing recommender system in tensorflow2. Algorithm: UserCF, ItemCF, LFM, SLIM, GMF, MLP, NeuMF, FM, DeepFM, MKR, RippleNet, KGCN and so on.
MIT License
410 stars 93 forks source link

您好，有一点想请教一下，关于MKR模型模型的输入中的n_user是user的id么？如果我想改成有没有别的好方法呢？ #11

Open ryouji62199 opened 3 years ago

SSSxCCC commented 3 years ago

用dict做一个映射就好了，输入模型的时候使用id，其它时候可以将id通过映射转换成你需要的对象。