Leo followed Machine Learning: Decision Trees sklearn of

sklearn framework

Mapping function

1.10. Decision Trees

1.10.1. Classification

from sklearn import tree
X = [[0, 0], [1, 1]]
Y = [0, 1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, Y)

1.10.2. Regression

from sklearn import tree
X = [[0, 0], [2, 2]]
y = [0.5, 2.5]
clf = tree.DecisionTreeRegressor()
clf = clf.fit(X, y)
clf.predict([[1, 1]])

source address

https://scikit-learn.org/stable/modules/tree.html