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
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