使用OpenKE的预训练实体嵌入

Question

在使用openke预训练好的实体嵌入时，我通过实体链接工作得到了具体的实体名称，但是在openke给出的entity2id文件中（每一行是一个实体和它对应的id，需要通过此id提取出实体的预训练嵌入），每一行的实体并非是实体名称，似乎是一个代号，例如Q1。我不理解这个代号的意思，也没有找到可以帮助将实体映射到这些代号的文件，我该如何使用我的实体名称获取它准确的id（在OpenKE的预训练嵌入中）？openke预训练嵌入的地址是openke

Answer 1

引自免费微信小程序：皆我百晓生

在使用OpenKE的预训练实体嵌入时，遇到的问题是实体名称与OpenKE提供的entity2id文件中的标识符不匹配。通常情况下，OpenKE使用的实体标识符遵循了Freebase的命名规则，例如“Q1”，这表示的是Freebase数据库中的一个唯一ID。如果你的数据集不是来自Freebase，那么你需要一个映射文件来将你的实体名称转换为OpenKE的实体ID。

解决这个问题的方法有两种：

1. 创建映射文件： 你需要一个映射文件，其中包含你的实体名称和它们对应的OpenKE ID。你可以手动创建这个文件，或者如果可能的话，从数据源中找到这样的映射。映射文件应该是这样的格式：

   实体名称    OpenKE ID
   entity_name_1    Q1
   entity_name_2    Q2
   ...
   

   然后，你需要读取这个映射文件，并建立一个字典，以便在后续处理中快速查找实体ID。

   这里有一个Python示例来帮助你完成这个任务：

   def load_mapping(filename):
       mapping = {}
       with open(filename, 'r') as f:
           for line in f:
               entity, id = line.strip().split('\t')
               mapping[entity] = id
       return mapping
   
   # 使用映射函数加载映射文件
   mapping_file = "your_mapping_file.txt"
   mapping = load_mapping(mapping_file)
   

2. 如果无法获得映射文件： 如果没有现成的映射文件，你可能需要尝试从Freebase或者其他知识图谱中寻找你的实体名称对应的ID。但是请注意，这种方法可能会非常耗时且不一定能找到所有实体的对应关系，因为不同知识图谱的实体命名并不总是兼容的。

一旦你有了实体名称到OpenKE ID的映射，你就可以根据这个映射来获取预训练的实体嵌入了。假设你已经加载了OpenKE的预训练嵌入到一个名为embeddings的字典中，其中键是OpenKE ID，值是对应的向量。下面是如何根据实体名称获取嵌入的例子：

def get_embedding_by_name(name, embeddings, mapping):
    id = mapping.get(name)
    if id is not None:
        return embeddings[id]
    else:
        print(f"找不到实体'{name}'的映射")
        return None

# 假设embeddings是预训练嵌入字典
embedding_for_entity = get_embedding_by_name("your_entity_name", embeddings, mapping)

确保你已经正确地加载了预训练的嵌入以及映射文件，然后使用上述方法，你应该能够根据实体名称获取其预训练嵌入。