import { DecisionTreeClassifier } from 'machinelearn/tree';
const features = ['color', 'diameter', 'label'];
const decision = new DecisionTreeClassifier({ featureLabels: features });

const X = [['Green', 3], ['Yellow', 3], ['Red', 1], ['Red', 1], ['Yellow', 3]];
const y = ['Apple', 'Apple', 'Grape', 'Grape', 'Lemon'];
decision.fit({ X, y });
decision.printTree(); // try it out yourself! =)

decision.predict({ X: [['Green', 3]] }); // [ 'Apple' ]
decision.predict({ X }); // [ [ 'Apple' ], [ 'Apple', 'Lemon' ], [ 'Grape', 'Grape' ], [ 'Grape', 'Grape' ], [ 'Apple', 'Lemon' ] ]

import { DecisionTreeClassifier } from 'machinelearn/tree';
const decision = new DecisionTreeClassifier({ featureLabels: null });

const X = [[0, 0], [1, 1]];
const Y = [0, 1];
decision.fit({ X, y });
decision2.predict({ row: [[2, 2]] }); // [ 1 ]