rocelib.models.sklearn_models package

Submodules

class rocelib.models.sklearn_models.TrainableDecisionTreeModel.TrainableDecisionTreeModel[source]

A Decision Tree Classifier model wrapper for scikit-learn.

Inherits from TrainableSKLearnModel and initializes a DecisionTreeClassifier as the underlying model.

class rocelib.models.sklearn_models.TrainableLogisticRegressionModel.TrainableLogisticRegressionModel[source]

A Logistic Regression Classifier model wrapper for scikit-learn.

Inherits from TrainableSKLearnModel and initializes LogisticRegression as the underlying model.

class rocelib.models.sklearn_models.TrainableSVMModel.TrainableSVMModel[source]

A SVM model wrapper for scikit-learn.

Inherits from TrainableSKLearnModel and initializes SVC as the underlying model.

class rocelib.models.sklearn_models.TrainableSKLearnModel.TrainableSKLearnModel(model)[source]

A base class for scikit-learn models.

This class wraps a scikit-learn model and provides methods for training, predicting, and evaluating the model. Inherits from TrainableModel.

train(X, y, **kwargs)[source]

Trains the scikit-learn model.

@param y: The target variable, should be a DataFrame.