search

hand10ryo / PyTorchCML

PyTorchCML is a library of PyTorch implementations of matrix factorization (MF) and collaborative metric learning (CML), algorithms used in recommendation systems and data mining.
MIT License
20 stars 2 forks source link

Combine the base trainer into one #26

Closed hand10ryo closed 2 years ago

hand10ryo commented 2 years ago

It is difficult to understand that there is a Base CML Trainer and a Base MF Trainer. The same trainer should be used for both CML and MF.