Source code for rocelib.models.sklearn_models.TrainableLogisticRegressionModel

from sklearn.linear_model import LogisticRegression

from rocelib.models.sklearn_models.TrainableSKLearnModel import TrainableSKLearnModel


[docs] class TrainableLogisticRegressionModel(TrainableSKLearnModel): """ A Logistic Regression Classifier model wrapper for scikit-learn. Inherits from TrainableSKLearnModel and initializes LogisticRegression as the underlying model. """ def __init__(self): super().__init__(LogisticRegression(solver='liblinear'))