pycaret model 到 pmml 的 sklearn2pmml 错误：object 不是 BaseEstimator 的实例

Question

使用 pycaret 创建 model 到目前为止没有问题（训练了一堆不同的模型）但是当使用 catboost 时，我无法保存到 pmml。 相同的代码适用于具有相同数据的 xgboost 和 lightgbm。

from sklearn2pmml.pipeline import PMMLPipeline
from sklearn2pmml import sklearn2pmml,make_pmml_pipeline
from pycaret.regression import setup,tune_model,finalize_model,create_model
    

clf=setup(data=df,
          target='target_Var',
          train_size= 0.8,
          fold_shuffle = True,
          fold = 5,
          fold_strategy="groupkfold",
          fold_groups="id",
          html = False,
          silent = True,
          session_id = 1,
          n_jobs = -1)

    model = create_model('catboost')
    tuned_model = tune_model(model, fold=5)
    final_model = finalize_model(tuned_model)
    model_pipeline = make_pmml_pipeline(final_model)
    sklearn2pmml(model_pipeline_pm, model_path)
        
             12     tuned_model = tune_model(model, fold=5)
             13     final_model = finalize_model(tuned_model)
        ---> 14     model_pipeline = make_pmml_pipeline(final_model)
        
        
        
        ~\Anaconda3\envs\cloned2\lib\site-packages\sklearn2pmml\__init__.py in make_pmml_pipeline(obj, active_fields, target_fields)
            138 
            139     """
        --> 140         steps = _filter_steps(_get_steps(obj))
            141         pipeline = PMMLPipeline(steps)
            142         if active_fields is not None:
        
        ~\Anaconda3\envs\cloned2\lib\site-packages\sklearn2pmml\__init__.py in _get_steps(obj)
             97                 return [("estimator", obj)]
             98         else:
        ---> 99                 raise TypeError("The object is not an instance of {0}".format(BaseEstimator.__name__))
            100 
            101 def _filter(obj):
        
        TypeError: The object is not an instance of BaseEstimator

Answer 1

您应该对 catbost model 使用特定的导出方法：

if model == 'catboost' :
        final_model.save_model(path,format="pmml")

Answer 2

请参阅 JPMML-SkLearn/SkLearn2PMML 堆栈支持的第 3 方包列表 - https://github.com/jpmml/jpmml-sklearn#supported-packages

从 SkLearn2PMML 0.84(.2) 开始，LightGBM 和 XGBoost 包被列为支持，而 CatBoost package 不支持。

CatBoost 具有内置的 PMML 转换器（请参阅“应用 pmml ”）。 尝试将其集成到您的工作流程中。