sklearn2pmml错误与pmml :对象不是BaseEstimator的实例

Question

创建一个使用pycaret的模型，到目前为止没有问题(训练了一组不同的模型)，但是当使用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模型使用特定的导出方法：

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

Answer 2

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

在SkLearn2PMML 0.84(.2)中，LightGBM和XGBoost包被列为受支持的包，而CatBoost包则不受支持。

CatBoost内置了PMML转换器(参见"应用pmml")。尝试将其集成到您的工作流中。