Catboost: как передать cat_features в сохраненную модель в Python? - PullRequest
0 голосов
/ 10 февраля 2020

У меня есть эта предварительно обученная сохраненная модель, в которой я сообщил свои категориальные особенности, и она работает нормально, если я прогнозирую сразу после тренировки. Теперь я хочу использовать его снова в другом контексте, но я не знаю, как правильно сообщить категориальные особенности. Я попытался это:

model = CatBoostClassifier(cat_features=var_categ)  
model.load_model('catmod.cat')

, но когда я пытаюсь предсказать:

modelo.predict(base)

Я получаю эту ошибку:

CatBoostError: features data: pandas.DataFrame column 'cod_var1' has dtype 'category' but is not in  cat_features list

Да, я дважды проверил столбец находится в var_categ.

1 Ответ

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

Прежде всего, вам не нужно указывать классификатор catboost cat_features, потому что модель уже содержит эту информацию из load_model.

Из вашей ошибки я могу догадаться, что при использовании предиката для нового набора данных, ваши функции смещены на 1, что дает вам ошибку.

...