xxxxxxxxxx
#for class names
y_class_names=list(df['your_class_type_column'].unique())
or you can use class_names=True, # see below
#for feature names
X_col_names = list(X_train.columns)
feature_names = X_col_names
or
X_names = list(df.columns[:-1])
eg:
import graphviz
# DOT data
dot_data = tree.export_graphviz(clf_gini, out_file=None,
feature_names=X_col_names,
class_names=True, # or class_names=y_class_names,
filled=True)
# Draw graph
graph = graphviz.Source(dot_data, format="png")
graph