AttributeError: Can‘t get attribute ‘WordVocab‘ on ＜module ‘__main__‘ from ‘genetic_algorithm.py‘＞

原因：

背景：在看代码的时候想看看一个pkl文件的文件结构和里面的数据，于是就用pickle.load打开，但是出现了AttributeError: Can't get attribute 'WordVocab' on <module '__main__'错误。

原因：就像我们在保存模型的时候有两种方式，一种是保存整个模型，虽然很大但是使用方便；另一种是只保存了模型的参数，这里加载时就需要先初始化一个模型。这里也是一样的，vocab.pkl在加载时也需要找到其依赖的类，而在报错信息中也说明了，需要导入的类是WordVocab。所以需要在项目中找到类WordVocab，然后在调用pickle.load('.../vocab.pkl','rb')的python文件中import该类。

报错代码：

@staticmethod
    def load_vocab(vocab_path: str) -> 'WordVocab':
        with open(vocab_path, "rb") as f:
            return pickle.load(f)

 

解决方法，在“genetic_algorithm.py”文件中导入“WordVocab”类

参考

pickle.load报错【AttributeError: Can‘t get attribute ‘Vocabulary‘ on ＜module ‘__main__‘】