1. 程式人生 > 其它 >Sklearn實現決策樹

Sklearn實現決策樹

技術標籤:機器學習# Sklearn筆記sklearn決策樹機器學習

>>> 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]])
array([ 0.5])