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

Графики важности признаков для анализа поступления в университет

Вы работаете в команде специалистов по данным в университете. Ваша задача — определить, какие факторы действительно влияют на решение о зачислении, а какие имеют меньшее значение. Приёмная комиссия знает, что средний балл (CGPA) играет ключевую роль, однако хочет это подтвердить и выявить другие важные факторы, влияющие на результат. С помощью модели RandomForestRegressor вы визуализируете важность признаков, чтобы чётко определить, какие характеристики абитуриентов наиболее значимы, а какие оказывают меньшее влияние на процесс принятия решений.

Библиотека shap и обучающие данные (X_train, y_train) уже загружены.

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

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

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

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

  • Вычислите shap_values с помощью TreeExplainer.
  • Используйте полученные shap_values, чтобы построить столбчатый график важности признаков и проанализировать его.

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

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

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

# Derive shap values
explainer = ____
shap_values = ____

# Plot the feature importance plot
____
Редактировать и запускать код