def createTree(train, train_labels):
clf = DecisionTreeClassifier(criterion="entropy",
splitter="best",
min_samples_leaf=12)
clf.classes = [1, 2, 3, 4, 5]
print("classes: ", clf.classes)
clf.fit(train, train_labels)
return clf
