Refefer / fastxml

FastXML / PFastXML / PFastreXML - Implementation of Extreme Multi-label Classification
Other
149 stars 47 forks source link

Update trainer.py #25

Closed siddu9501 closed 4 years ago

siddu9501 commented 4 years ago

Changing to max_iter (or removing n_iter altogether) fixes the problem with SGDClassifier attributes. n_iter is depricated and was removed starting scikit-learn 0.21

You might also consider changing the requirements file to lock the versions to exactly what the repo works with instead of having a "library>=version " style.