Machine Learning Tutorial

Decision Tree for Classification

from sklearn.tree import DecisionTreeClassifier

tree = DecisionTreeClassifier(max_depth=4, random_state=0)