Вычисление важности признаков с помощью деревьев решений
Вы построили классификатор на основе дерева решений для выявления пациентов с риском сердечно-сосудистых заболеваний, используя соответствующий набор данных. Теперь необходимо объяснить модель, проанализировав важность признаков, чтобы определить ключевые факторы предсказания болезни сердца и обеспечить более целенаправленные медицинские вмешательства.
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()