text_plain_output_1.png
from sklearn.tree import DecisionTreeClassifier

drugTree = DecisionTreeClassifier(criterion='entropy', max_depth=4)

drugTree.fit(x_train, y_train)

y_predict = drugTree.predict(x_test)
y_predict