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.