Как построить график для признаков xgboost?

Есть данные для задачи классификации, использовал xgboost classifier для того, чтобы воспользоваться feature_importance_, чтобы в дальнейшем получить признаки которые сильно влияют на выходные данные (в моей задачи это 0 и 1). Написал следующий код:

import re
regex = re.compile(r"\[|\]|<", re.IGNORECASE)
df.columns = [regex.sub("_", col) if any(x in str(col) for x in set(('[', ']', '<'))) else col for col in df.columns]
X,y = df.drop('class', axis=1), df['class'].copy()
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=7)
xgb_classifier.fit(X_train,y_train)

Теперь интересует вопрос, как получить график, который находится на картинке. Чтобы было отображено, какой из признаков влияет на class = 0, а какой на class = 1.