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

Early stop for overfitting? #42

Open shivamtundele opened 2 years ago

shivamtundele commented 2 years ago

I tried find a method inside the model and trainer methods for early stopping. Is there a way?

hand10ryo commented 2 years ago

Sorry for the delay in responding. I recognize that this is a priority improvement issue, although it is not currently being considered. If you know of a better way to do this, could you please make a PR? I'd be happy to share the design of the implementation!