Есть ли какие-то способы узнать, на какой подвыборке данных было установлено дерево XGBoost? - PullRequest
2 голосов
/ 11 февраля 2020

Я занимаюсь практикой XGBoost и хотел бы знать, на каком подмножестве данных были установлены деревья XGBRegressor. Вот список параметров, которые я использую:

params = {'learning_rate': 0.09, 
          'n_estimators': 5,
          'objective': 'reg:squarederror',
          'max_depth': 6,
          'subsample': 0.2 
         }

Я работаю с Python 3.7 и библиотекой XGBoost. Так есть ли способ узнать это? Я прочитал do c, но не смог найти ничего полезного.

1 Ответ

0 голосов
/ 11 февраля 2020

Вы можете легко нарисовать дерево и проверить это, вы можете сделать это с помощью

export_graphiviz(uralgo,"filename",feature_names,rounded=True)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...