Нужна помощь в настройке ширины строки и размера графика для горизонтальной гистограммы с помощью Matplotlib - PullRequest
0 голосов
/ 04 февраля 2020

Вот мой код, но я хочу отрегулировать ширину строки, чтобы имена были понятными для чтения, и отрегулируйте размер графика. Любая помощь будет принята с благодарностью.

# Building the model 
extra_tree_forest = ExtraTreesClassifier(n_estimators = 81, 
                                        criterion ='entropy', max_features = 5) 

# Training the model 
extra_tree_forest.fit(X, y) 

# Computing the importance of each feature 
feature_importance = extra_tree_forest.feature_importances_ 

# Normalizing the individual importances 
feature_importance_normalized = np.std([tree.feature_importances_ for tree in 
                                        extra_tree_forest.estimators_], 
                                        axis = 0) 

# Plotting a Bar Graph to compare the models 
plt.barh(X.columns, feature_importance_normalized) 
plt.xlabel('Feature Labels') 
plt.ylabel('Feature Importances') 
plt.title('Comparison of Different Feature Importances (Cluster1)') 
plt.show()

enter image description here

1 Ответ

0 голосов
/ 04 февраля 2020

Вам просто нужно увеличить размеры вашей фигуры. Добавьте следующую строку вверху вашего кода

plt.figure(figsize=(<width>, <height>)) # dimensions are in inches

В вашем случае было бы разумно выбрать пропорцию, пропорциональную количеству столбцов в вашем наборе данных

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...