You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

咨询:使用pickle.load()加载PMML模型管道的函数返回值类型提示

为pickle加载PMML模型管道的函数添加类型提示

根据你使用的PMML模型管道的具体实现,返回类型提示可以按以下几种情况处理:

  • 如果你的PMML模型管道是用sklearn2pmml库构建的,返回类型应指定为PMMLPipeline:
    先导入对应类,再修改函数的返回类型提示,同时注意修正原代码中写死文件路径的问题:

    from sklearn2pmml.pipeline import PMMLPipeline
    import pickle
    
    def load_model(model_path: str) -> PMMLPipeline:
        with open(model_path, "rb") as f:
            model_pipeline = pickle.load(f)
    
        return model_pipeline
    
  • 如果是自定义的PMML模型管道类,直接将返回类型指定为你自定义的类即可。

  • 若暂时无法确定具体类型,或模型可能是多种PMML相关类型的混合,可以用typing.Any作为临时方案,但这种方式无法发挥类型检查的优势,不推荐长期使用:

    from typing import Any
    import pickle
    
    def load_model(model_path: str) -> Any:
        with open(model_path, "rb") as f:
            model_pipeline = pickle.load(f)
    
        return model_pipeline
    

内容的提问来源于stack exchange,提问作者wawawa

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.07.26 01:40:00