跟着Leo机器学习：sklearn之Decision Trees

sklearn框架

函数导图

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]])

源地址

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