Decision tree Classifier (sklearn)

dtclf.py

from sklearn.datasets import load_iris #load dataset from scikitlearn
                                       #This is a function which will return X and Y
from sklearn import tree

iris = load_iris()   #load_iris() returns iris.data (x)  and iris.target (Y)

clf = tree.DecisionTreeClassifier()   #we create a Decision Tree Classifier from sklearn.tree
clf.fit(iris.data, iris.target)     #Training

print("input: ", [4.9, 3.4, 2, 0.4])     #predictions
print("Prrediction: ", clf.predict([[4.9, 3.4, 2, 0.4]]))