API Reference

The public API is re-exported from the top-level sgtlearn package.

SGTClassifier

Shape Generalized Tree classifier.

SGTRegressor

Shape Generalized Tree regressor.

RandomSGForestClassifier

Random forest of Shape Generalized Trees for classification.

RandomSGForestRegressor

Random forest of Shape Generalized Trees for regression.

plot_tree

Render a fitted SGT estimator with matplotlib (see module docstring).

configure_feature_dict

Resolve logical features for tree training.

export_graphviz

Serialize the fitted tree as Graphviz DOT (not implemented).

export_text

Return a human-readable multiline description of the tree (not implemented).

make_plus

Generate the "Plus Sign" dataset.

tao.TAO_refine

Refine a fitted shape-generalized tree or forest in place with TAO.