DecisionTreeClassifier
API reference for DecisionTreeClassifier
Tree.DecisionTreeClassifier
interface DecisionTreeProps {
max_depth?: number;
min_samples_split?: number;
criterion?: 'entropy' | 'gini';
}
constructor(props: DecisionTreeProps = {})
Example
const dt = new DecisionTreeClassifier({ criterion: 'gini' });
dt.fit(X, Y);
const result = dt.predict(T);