НачатьНачать бесплатно

Вычисление важности признаков с помощью деревьев решений

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

matplotlib.pyplot импортирован как plt. Переменные X_train и y_train уже загружены.

Это упражнение является частью курса

Объяснимый ИИ на Python

Посмотреть курс

Инструкции к упражнению

  • Извлеките значения важности признаков из объекта model.
  • Постройте график feature_importances для заданных feature_names.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

model = DecisionTreeClassifier(random_state=42)
model.fit(X_train, y_train)

# Derive feature importances
feature_importances = ____
feature_names = X_train.columns

# Plot the feature importances
____
plt.show()
Редактировать и запускать код