entity_linking_en.json

{
  "chainer": {
    "in": ["entity_substr", "tags", "sentences", "entity_offsets", "sentences_offsets"],
    "pipe": [
      {
        "class_name": "torch_transformers_entity_ranker_infer",
        "id": "entity_descr_ranking",
        "pretrained_bert": "{TRANSFORMER}",
        "encoder_weights_path": "{MODELS_PATH}/entity_linking_eng/encoder.pth.tar",
        "bilinear_weights_path": "{MODELS_PATH}/entity_linking_eng/bilinear.pth.tar",
        "special_token_id": 30522,
        "emb_size": 512,
        "block_size": 8
      },
      {
        "class_name": "entity_linker",
        "in": ["entity_substr", "tags", "sentences", "entity_offsets", "sentences_offsets"],
        "out": ["entity_ids", "entity_conf", "entity_pages"],
        "load_path": "{DOWNLOADS_PATH}/entity_linking_eng",
        "entities_database_filename": "el_eng.db",
        "entity_ranker": "#entity_descr_ranking",
        "rank_in_runtime": true,
        "num_entities_for_bert_ranking": 20,
        "use_gpu": false,
        "include_mention": false,
        "num_entities_to_return": 3,
        "lemmatize": true,
        "use_descriptions": true,
        "wikidata_file": "{DOWNLOADS_PATH}/wikidata/wikidata_lite.hdt",
        "use_connections": true,
        "use_tags": true,
        "full_paragraph": true,
        "return_confidences": true,
        "lang": "ru"
      }
    ],
    "out": ["entity_ids", "entity_conf", "entity_pages"]
  },
  "metadata": {
    "variables": {
      "ROOT_PATH": "~/.deeppavlov",
      "DOWNLOADS_PATH": "{ROOT_PATH}/downloads",
      "MODELS_PATH": "{ROOT_PATH}/models",
      "TRANSFORMER": "prajjwal1/bert-small"
    },
    "download": [
      {
        "url": "http://files.deeppavlov.ai/deeppavlov_data/entity_linking/el_db_eng.tar.gz",
        "subdir": "{DOWNLOADS_PATH}/entity_linking_eng"
      },
      {
        "url": "http://files.deeppavlov.ai/deeppavlov_data/entity_linking/el_ranker_eng.tar.gz",
        "subdir": "{MODELS_PATH}/entity_linking_eng"
      },
      {
        "url": "http://files.deeppavlov.ai/kbqa/wikidata/wikidata_lite.tar.gz",
        "subdir": "{DOWNLOADS_PATH}/wikidata"
      }
    ]
  }
}