Top2Vec v1.0.34模型Pickle序列化失败问题求助

问题确认 #

这个问题确实有其他用户遇到过，在Top2Vec的社区讨论中已有相关反馈，核心原因是v1.0.34版本中新增的Loader._recreate_base_user_object方法内定义了本地类_UserObject，而pickle无法序列化这类本地作用域内的对象。

可行解决办法 #

1. 使用官方内置的模型保存/加载方法（推荐）

Top2Vec本身提供了专门的持久化接口，完全可以替代pickle，且不存在兼容性问题：

# 保存模型
model.save("your_project_model")

# 加载模型
from top2vec import Top2Vec
model = Top2Vec.load("your_project_model")

2. 降级至稳定版本

如果必须依赖pickle序列化，可以临时降级到v1.0.33版本，该版本未引入导致序列化失败的代码逻辑：

pip install top2vec==1.0.33

3. 自定义参数序列化（临时 workaround）

若特殊场景下必须用pickle，可以手动提取模型的核心参数（如嵌入向量、主题向量等）进行序列化，之后再通过这些参数重建模型：

# 提取核心参数
model_params = {
    "embeddings": model.embeddings,
    "document_ids": model.document_ids,
    "topic_vectors": model.topic_vectors,
    "topic_sizes": model.topic_sizes,
    "topic_nums": model.topic_nums
}

# 序列化参数
with open("model_params.pickle", "wb") as file:
    pickle.dump(model_params, file)

# 加载参数并重建模型
with open("model_params.pickle", "rb") as file:
    loaded_params = pickle.load(file)

from top2vec import Top2Vec
rebuilt_model = Top2Vec(
    embeddings=loaded_params["embeddings"],
    document_ids=loaded_params["document_ids"]
)
# 手动赋值其他参数
rebuilt_model.topic_vectors = loaded_params["topic_vectors"]
rebuilt_model.topic_sizes = loaded_params["topic_sizes"]
rebuilt_model.topic_nums = loaded_params["topic_nums"]

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