Следующий код обучает классификатор дерева решений для некоторых данных, строит дерево решений в формате DOT и отображает его. Единственная проблема: дерево довольно большое, и я вижу только половину. Как я могу изменить размер фигуры, отображаемой в моем блокноте jupyter? Кажется, что функция display(.)
не принимает параметры размера.
# Features
features = ['f1','f2','f3'] # removed 'ID'
X = train[features]
# Target
y = train['response']
# Split dataset into training set and test set
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=1) # 70% training and 30% test
# Train Decision Tree Classifer
clf = DecisionTreeClassifier(random_state=0)
clf = clf.fit(X_train,y_train)
#Predict the response for test dataset
y_pred = clf.predict(X_test)
# Visualize the tree (tree is too big)
from IPython.display import display
display(graphviz.Source(export_graphviz(clf,out_file=None,feature_names=features,class_names=True)))