Method matrix

This table summarizes which API methods are implemented by each predictor.

Method ModalBoundaryClustering ShuShu CheChe ModalScoutEnsemble InsideForest Brief description
fit ✓ ✓ ✓ ✓ ✓ Fit the model to the dataset.
fit_predict ✓ ✓ ✓ ✓ — Train and return assigned labels or regions.
fit_transform ✓ ✓ ✓ ✓ — Fit and transform data into an internal representation.
transform ✓ ✓ ✓ ✓ ✓ Transform new samples according to the trained model.
predict ✓ ✓ ✓ ✓ ✓ Predict labels or values for new samples.
predict_proba ✓ ✓ ✓ ✓ — Estimate prediction probabilities or confidence.
decision_function ✓ ✓ ✓ ✓ — Return score or distance to the decision boundary.
predict_regions ✓ ✓ ✓ ✓ ✓ Indicate the region or cluster each sample belongs to. Similar to get_cluster in some predictors.
score ✓ ✓ ✓ ✓ — Compute a performance metric on test data.
save ✓ ✓ ✓ ✓ ✓ Save the trained model to disk.
load ✓ ✓ ✓ ✓ — Load a previously saved model.
plot_pairs ✓ ✓ ✓ ✓ — Plot feature pairs with regions or clusters.
plot_pair_3d ✓ ✓ ✓ ✓ — Plot two features and the response in 3D.
interpretability_summary ✓ ✓ ✓ ✓ ✓ Interpretability summary of regions or the model.

See the README for more context.