self.ent_inlinks = config['entity_inlinks'] KeyError: 'entity_inlinks

[Jorigorn]

python main.py --mode train --order offset --model_path model --method RL 是可以执行成功的。 但是eval 的时候报错。

python main.py --mode eval --order offset --model_path model --method RL

root@:/data/DCA$ ~/.conda/envs/keras_bert/bin/python main.py --mode eval --order offset --model_path model --method RL load conll at ../data/generated/test_train_data load csv 370United News of India process coref load conll reorder mentions within the dataset create model --- create EDRanker model --- prerank model --- create NTEE model --- --- create AbstractWordEntity model --- main model try loading model from model --- create MulRelRanker model --- --- create LocalCtxAttRanker model --- --- create AbstractWordEntity model --- Traceback (most recent call last): File "main.py", line 204, in ranker = EDRanker(config=config) File "../DCA/ed_ranker.py", line 52, in init self.model = load_model(self.args.model_path, ModelClass) File "../DCA/abstract_word_entity.py", line 28, in load model = model_class(config) File "../DCA/mulrel_ranker.py", line 32, in init self.ent_inlinks = config['entity_inlinks']

data 目录下是这样的： ll data total 135990 drwxrwxr-x 1 mqq mqq 108039782 Nov 6 2018 basic_data drwxrwxr-x 1 mqq mqq 6472491 Nov 6 2018 data -rw-rw-r-- 1 mqq mqq 1245609 Aug 20 2019 doc2type.pkl -rw-rw-r-- 1 mqq mqq 120266616 May 18 2019 ent2desc.json -rw-rw-r-- 1 mqq mqq 9864329 Aug 20 2019 entity2type.pkl -rw-rw-r-- 1 mqq mqq 7869649 Aug 20 2019 entityid_dictid_inlinks_uniq.pkl drwxrwxr-x 1 mqq mqq 2251141855 Jan 23 2019 generated -rw-rw-r-- 1 mqq mqq 5862 Aug 20 2019 stopwords-multi.txt -rw-rw-r-- 1 mqq mqq 82 Aug 20 2019 symbols.txt drwxrwxr-x 1 mqq mqq 21420384 Nov 6 2018 test_data drwxrwxr-x 1 mqq mqq 898623014 Nov 6 2018 word_ent_embs

麻烦看下，谢谢。