Catboost：如何將 cat_features 傳遞給 Python 中保存的模型？

Question

我有這個預先訓練好的保存模型，我在其中告知了我的分類特征，如果我在訓練后立即進行預測，它就可以正常工作。 現在我想在另一個上下文中再次使用它，但我不知道如何正確地告知分類特征。 我試過這個：

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 中。

Answer 1

首先，你不需要指定 catboost 分類器 cat_features，因為模型已經有來自 load_model 的這個信息。

我會從你的錯誤中猜測，當你在新數據集上使用 predict 時，你的特征移動了 1 個位置，從而給你錯誤。