search

LiJunnan1992 / DivideMix

Code for paper: DivideMix: Learning with Noisy Labels as Semi-supervised Learning
MIT License
543 stars 84 forks source link

Hyperparameter setting of GMM #20

Closed RumengYi closed 4 years ago

RumengYi commented 4 years ago

Hello, thanks for your excellent work! Can two input losses be entered in the fit() of GMM? For example, gmm.fit(input_loss1, input_loss2)? I am curious and want to try to use two losses to model GMM.

LiJunnan1992 commented 4 years ago

Hi, thanks for your interest in our method! You can concatenate the two losses and pass into fit(), please see here for how to use gmm.fit(): https://scikit-learn.org/stable/modules/generated/sklearn.mixture.GaussianMixture.html#sklearn.mixture.GaussianMixture.fit

RumengYi commented 4 years ago

Thank you very much! Could you please provide the reproduced code of F-correction?

LiJunnan1992 commented 4 years ago

Hi, you can check out the author's implementation here: https://github.com/giorgiop/loss-correction