Я обучил и проверил модель XGBoost на компьютере с графическим процессором и протравил ее. Когда я загружаю рассол и тестирую на одной машине, он работает отлично. Я получаю мультикласс ROC_AUC и работает. При тестировании на другой машине, на которой есть только резервные процессоры, я загружаю рассол и при необходимости устанавливаю:
xgboost_clf = xgboost_clf.set_params(predictor = 'cpu_predict')
xgboost_clf = xgboost_clf.set_params(tree_method = 'hist')
Я запускаю это:
Y_test_pred_xgboost = xgboost_clf.predict(X_test)
Но все результаты СЕЙЧАС принадлежат первому из моих нескольких классов. Я был бы очень признателен за ответ на вопрос: что я еще не учел?
С уважением,
Яан:)
PS! Кажется, не решает эту проблему, которая является фиксированной проблемой.
https://github.com/h2oai/h2o4gpu/issues/625