КМ.10/24.6/11.10/11.Задача 1

Сформируйте таблицу важности признаков на данных о пассажирских перевозках:

1. обучите дерево решений;
2. составьте таблицу важности признаков;
3. отсортируйте таблицу от самых важных признаков к менее важным.

# импортируем библиотеки и объявляем константы
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import OneHotEncoder
from sklearn.tree import DecisionTreeClassifier

RANDOM_STATE = 42

# подготавливаем данные заранее созданной функцией
X_train, X_test, y_train, y_test = prepare_data('train_satisfaction.csv')

# инициализируем модель дерева решений с лучшими гиперпараметрами
model = DecisionTreeClassifier(
    max_depth=4,
    min_samples_leaf=1,
    min_samples_split=2,
    random_state=RANDOM_STATE
)

# обучите модель
model.fit(X_train, y_train)

# сформируйте таблицу важности признаков
feature_importances = pd.DataFrame(
    {
        'Feature': X_train.columns,
        'Importance': model.feature_importances_
    })

# отсортируйте таблицу от большего значения важности к меньшему и выведите результат
print(feature_importances.sort_values(by='Importance', ascending=False))