【ML】3 决策树

from sklearn.datasets import load_iris
from sklearn import tree

iris = load_iris()

clf = tree.DecisionTreeClassifier()

clf = clf.fit(iris.data, iris.target)

print(clf.predict(iris.data[:1, :]))

from sklearn.datasets import load_iris
from sklearn import tree

iris = load_iris()

rgclf = tree.DecisionTreeRegressor()

rgclf = rgclf.fit(iris.data[2:], iris.target[2:])

print(rgclf.predict(iris.data[:2, :]))

print(iris.target[:2])