Графики важности признаков для анализа поступления в университет
Вы работаете в команде специалистов по данным в университете. Ваша задача — определить, какие факторы действительно влияют на решение о зачислении, а какие имеют меньшее значение. Приёмная комиссия знает, что средний балл (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
____