bioniumx.modeling.BaselineRFModel

class bioniumx.modeling.BaselineRFModel(n_estimators=100, random_state=42)[source]
__init__(n_estimators=100, random_state=42)[source]

Multilabel Random Forest Classifier for Biosignatures.

Methods

__init__([n_estimators, random_state])

Multilabel Random Forest Classifier for Biosignatures.

evaluate(X_test, y_test)

load(filepath)

predict(X_test)

predict_proba(X_test)

Returns an array of shape (n_samples, n_classes) containing probabilities.

save(filepath)

train(X_train, y_train)

Train the model.