Pickle反序列化报错：无法获取__main__模块的EnsembleModel属性

问题根源 #

pickle序列化对象时，会记录类的模块路径。如果当初你是在__main__模块（比如直接在临时脚本/交互环境里定义的EnsembleModel）中保存的模型，pickle会把类的路径标记为__main__.EnsembleModel。现在你把类移到了my_package.model包模块下，反序列化时pickle仍会尝试从__main__模块查找该类，自然找不到，就抛出了这个错误。

解决方法 #

方法1：重新规范序列化流程（推荐）

在正确的包模块路径下重新序列化模型，让pickle记录正确的类路径：
编写专门的保存脚本（比如save_model.py），从包中导入类后再保存：

# save_model.py
from my_package.model import EnsembleModel
import pickle

# 初始化、训练模型
model = EnsembleModel(...)
model.train(...)

# 序列化保存
with open("ensemble-model.mdl", "wb") as f:
    pickle.dump(model, f)

这样保存的模型，pickle会记录类路径为my_package.model.EnsembleModel，后续在main.py中通过MyModel加载时就能正常匹配。

方法2：临时映射模块路径（适合无法重新序列化的场景）

如果不能重新生成序列化文件，可以在反序列化前，把__main__模块的EnsembleModel指向包中的正确类：
在main.py开头添加以下代码：

# main.py
import sys
from my_package.model import EnsembleModel

# 将__main__模块的EnsembleModel指向包内的类
sys.modules['__main__'].EnsembleModel = EnsembleModel

# 执行加载逻辑
from my_package.model import MyModel
model = MyModel.load("ensemble-model.mdl")

这样pickle在查找__main__.EnsembleModel时，会找到我们映射过去的正确类。

方法3：封装处理逻辑到MyModel中

如果MyModel是负责加载的通用类，可以把路径映射逻辑封装到它的加载方法里：

# my_package/model.py
import pickle
import sys

class EnsembleModel:
    # 你的集成模型实现代码
    pass

class MyModel:
    @staticmethod
    def load(file_path):
        # 临时映射__main__中的EnsembleModel到当前模块的类
        if 'EnsembleModel' not in sys.modules['__main__'].__dict__:
            sys.modules['__main__'].EnsembleModel = EnsembleModel
        
        with open(file_path, "rb") as f:
            return pickle.load(f)

这种方式把处理逻辑藏在包内部，外部调用MyModel.load()时无需额外操作。

内容的提问来源于stack exchange，提问作者frisko