@kanaries/ml

ExtraTreeClassifier

API reference for ExtraTreeClassifier

Tree.ExtraTreeClassifier

interface ExtraTreeProps {
    max_depth?: number;
    min_samples_split?: number;
    criterion?: 'entropy' | 'gini';
    max_features?: number;
}

constructor(props: ExtraTreeProps = {})

Example

const clf = new ExtraTreeClassifier();
clf.fit(X, Y);
const result = clf.predict(T);