咨询:使用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
相关产品推荐
相关产品推荐

