flexs.baselines.models.sklearn_models¶

Define scikit-learn model wrappers as well a few convenient pre-wrapped models.

class flexs.baselines.models.sklearn_models.LinearRegression(alphabet, **kwargs)[source]¶

Sklearn linear regression.

class flexs.baselines.models.sklearn_models.LogisticRegression(alphabet, **kwargs)[source]¶

Sklearn logistic regression.

class flexs.baselines.models.sklearn_models.RandomForest(alphabet, **kwargs)[source]¶

Sklearn random forest regressor.

class flexs.baselines.models.sklearn_models.SklearnClassifier(model, alphabet, name)[source]¶

Class for sklearn classifiers (uses model.predict_proba(…)[:, 1]).

class flexs.baselines.models.sklearn_models.SklearnModel(model, alphabet, name)[source]¶

Base sklearn model wrapper.

train(sequences, labels)[source]¶: Flatten one-hot sequences and train model using model.fit.

class flexs.baselines.models.sklearn_models.SklearnRegressor(model, alphabet, name)[source]¶

Class for sklearn regressors (uses model.predict).