yao8839836 / kg-bert

KG-BERT: BERT for Knowledge Graph Completion
Apache License 2.0
679 stars 141 forks source link

bert-base-uncased文件 #16

Open 1eclipse opened 4 years ago

1eclipse commented 4 years ago

这个文件自己从谷歌下载,直接放在项目下可以用吗,需不需要处理三个.cpkt文件呢? 谢谢

yao8839836 commented 4 years ago

@1eclipse 你好,我下载的是pytorch模型:

'bert-base-uncased': "https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-uncased.tar.gz",
'bert-large-uncased': "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-uncased.tar.gz",
'bert-base-cased': "https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-cased.tar.gz",
'bert-large-cased': "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased.tar.gz",
'bert-base-multilingual-uncased': "https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-multilingual-uncased.tar.gz",
'bert-base-multilingual-cased': "https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-multilingual-cased.tar.gz",
'bert-base-chinese': "https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-chinese.tar.gz",

可以解压后在代码里指定目录,目录中是这三个文件bert_config.json pytorch_model.bin vocab.txt

1eclipse commented 4 years ago

恩,感谢您!

csq0226 commented 1 year ago

@yao8839836 您好,请问解压后的模型名字pytorch_model需要改成bert-base-cased吗?我运行之后一直在加载vacab.txt,没有其他反应了

yao8839836 commented 1 year ago

@csq0226 您好,应该不需要改pytorch_model文件名

csq0226 commented 1 year ago

@yao8839836 感谢,已运行成功!还有个问题想请教一下您,关系预测、链路预测的结果要怎样保存或者显示出来呢?这个运行完之后只有评估的结果,并没有看到对知识图谱进行补全的结果?

yao8839836 commented 1 year ago

@csq0226 您好,可以定一个阈值,得分超过阈值的三元组作为补充的三元组,加入现有知识图谱的三元组集合。