import pandas as pd from pandas import DataFrame from sklearn.tree import DecisionTreeClassifier from sklearn.model_selection import train_test_split from sklearn