7/21/2020 DecisionTree Classifier.

ipynb - Colaboratory

from sklearn import tree

X = [[0,0],[1,1]]
Y = [0,1]
dt = tree.DecisionTreeClassifier()
dt = dt.fit(X,Y)
dt.predict([[2.,2.]])

array([1])

dt.predict_proba([[2.,2.]])

array([[0., 1.]])

tree.plot_tree(dt)

[Text(167.4, 163.07999999999998, 'X[0] <= 0.5\ngini = 0.5\nsamples = 2\nvalue = [1, 1

Text(83.7, 54.360000000000014, 'gini = 0.0\nsamples = 1\nvalue = [1, 0]'),
Text(251.10000000000002, 54.360000000000014, 'gini = 0.0\nsamples = 1\nvalue = [0, 1

https://colab.research.google.com/drive/1rhaY0WDIxgzu9Ixnn88LJcLZBVt2rvry#scrollTo=nqWH8_0C-ZBC&printMode=true 1/2
7/21/2020 DecisionTree Classifier.ipynb - Colaboratory

https://colab.research.google.com/drive/1rhaY0WDIxgzu9Ixnn88LJcLZBVt2rvry#scrollTo=nqWH8_0C-ZBC&printMode=true 2/2