Перенести загруженную модель из файла маринада - PullRequest
0 голосов
/ 24 октября 2018

У меня есть три набора данных: поезд , проверка , тест , и в настоящее время я использую классификатор XGBoost для выполнения задачи классификации.

Я обучил XGBClassifier на наборе train и сохранил его как файл pickle , чтобы избежать необходимости каждый раз переучивать его.Как только я загружаю модель из файла pickle, я могу использовать метод прогнозирования из него, но, похоже, я не могу обучить эту модель на наборе проверки или любом другом новом наборе данных.

Примечание: я не получаю никаких сообщений об ошибках, лабораторная ячейка jupyter выглядит так, как будто она работает отлично, но все ядра моего процессора отдыхают во время работы этой ячейки, поэтому я вижу, что модель не устанавливается.

Возможноэто может быть проблема с XGBoost или модели с дампами, которые не могут быть установлены снова после загрузки?

1 Ответ

0 голосов
/ 28 октября 2018

У меня был точно такой же вопрос год назад, Вы можете найти здесь вопрос и ответ

Хотя, таким образом, вы будете продолжать добавлять «деревья» (бустеры) квашей существующей модели с использованием ваших новых данных.

Возможно, лучше обучить новую модель вашим наборам данных обучения + проверки.

Что бы вы ни решили сделать, вы должны попробовать оба варианта иоцените свои результаты, чтобы увидеть, что лучше подходит для ваших данных.

...