mars-project / mars

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.
https://mars-project.readthedocs.io
Apache License 2.0
2.7k stars 325 forks source link

Add support for LogisticRegression #2734

Open wuyeguo opened 2 years ago

wuyeguo commented 2 years ago

Is your feature request related to a problem? Please describe. Implements LogisticRegression for Logistic Regression (aka logit, MaxEnt) classifier: API Like:https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html the param i need:

qinxuye commented 2 years ago

Perhaps not all solvers can be implemented as a distributed algorithm, do you need all solvers or just the one can work?

wuyeguo commented 2 years ago

just lbfgs is ok, if saga can be implemented is better

qinxuye commented 2 years ago

class_weight and l1_ratio are not supported by now.

The other parts look good to work, we will try to implement the absent algo and optimize the performance.