AnthonyMRios / leml

Method for large-scale multi-label classification
23 stars 3 forks source link

Predict predict proba #4

Closed Jopepato closed 5 years ago

Jopepato commented 5 years ago

I have added a new function named predict_proba that now returns the predictions of predict. Using predict as a function for returning the labels, 0 or 1. Being it more scikit-learn related :)

Kind regards ^^

AnthonyMRios commented 5 years ago

Thanks! Looks good to me.