├── .gitignore ├── LICENSE ├── README.md ├── bin ├── cloze_task_example.py ├── combine_wordnet_embeddings.py ├── create_pretraining_data_for_bert.py ├── evaluate_mrr.py ├── evaluate_perplexity.py ├── evaluate_wiki_linking.py ├── evaluate_wsd_official.py ├── extract_wordnet.py ├── preprocess_wsd.py ├── run_hyperparameter_seeds.sh ├── semeval2010_task8_scorer-v1.2.pl ├── tacred_scorer.py ├── write_semeval2010_task8_for_official_eval.py ├── write_tacred_for_official_scorer.py └── write_wic_for_codalab.py ├── kb ├── __init__.py ├── bert_pretraining_reader.py ├── bert_tokenizer_and_candidate_generator.py ├── bert_utils.py ├── common.py ├── dict_field.py ├── entity_linking.py ├── evaluation │ ├── __init__.py │ ├── classification_model.py │ ├── exponential_average_metric.py │ ├── fbeta_measure.py │ ├── semeval2010_task8.py │ ├── tacred_dataset_reader.py │ ├── tacred_predictor.py │ ├── ultra_fine_reader.py │ ├── weighted_average.py │ └── wic_dataset_reader.py ├── include_all.py ├── kg_embedding.py ├── kg_probe_reader.py ├── knowbert.py ├── knowbert_utils.py ├── metrics.py ├── multitask.py ├── self_attn_bucket_iterator.py ├── span_attention_layer.py ├── testing.py ├── wiki_linking_reader.py ├── wiki_linking_util.py └── wordnet.py ├── requirements.txt ├── setup.py ├── tests ├── __init__.py ├── evaluation │ ├── test_semeval2010_task8.py │ ├── test_simple_classifier.py │ ├── test_tacred_reader.py │ ├── test_ultra_fine_reader.py │ └── test_wic_reader.py ├── fixtures │ ├── bert │ │ ├── bert_config.json │ │ ├── bert_test_fixture.tar.gz │ │ ├── vocab.txt │ │ └── vocab_dir_with_entities_for_tokenizer_and_generator │ │ │ ├── entity.txt │ │ │ ├── non_padded_namespaces.txt │ │ │ └── tokens.txt │ ├── bert_pretraining │ │ └── shard1.txt │ ├── evaluation │ │ ├── semeval2010_task8 │ │ │ ├── semeval2010_task8.json │ │ │ └── vocab_entity_markers.txt │ │ ├── ultra_fine │ │ │ ├── train.json │ │ │ └── vocab.txt │ │ └── wic │ │ │ ├── train.data.txt │ │ │ ├── train.gold.txt │ │ │ └── vocab_entity_markers.txt │ ├── kg_embeddings │ │ ├── tucker_wordnet │ │ │ ├── model.tar.gz │ │ │ └── vocabulary │ │ │ │ ├── entity.txt │ │ │ │ ├── non_padded_namespaces.txt │ │ │ │ └── relation.txt │ │ ├── wn18rr_dev.txt │ │ └── wn18rr_train.txt │ ├── kg_probe │ │ └── file1.txt │ ├── linking │ │ ├── aida.txt │ │ ├── entities_universe.txt │ │ └── priors.txt │ ├── multitask │ │ ├── ccgbank.txt │ │ └── conll2003.txt │ ├── tacred │ │ ├── LDC2018T24.json │ │ └── vocab.txt │ ├── wordnet │ │ ├── cat_hat_mask_null_embedding.hdf5 │ │ ├── cat_hat_synset_mask_null_vocab.txt │ │ ├── cat_hat_vocabdir │ │ │ ├── entity.txt │ │ │ ├── non_padded_namespaces.txt │ │ │ └── tokens.txt │ │ ├── entities_cat_hat.jsonl │ │ ├── entities_fixture.jsonl │ │ └── wsd_dataset.json │ └── wordnet_wiki_vocab │ │ ├── entity_wiki.txt │ │ ├── entity_wordnet.txt │ │ └── non_padded_namespaces.txt ├── test_bert_pretraining_reader.py ├── test_bert_tokenizer_and_candidate_generator.py ├── test_common.py ├── test_dict_field.py ├── test_entity_linking.py ├── test_kg_embedding.py ├── test_kg_probe_reader.py ├── test_knowbert.py ├── test_metrics.py ├── test_multitask.py ├── test_self_attn_iterator.py ├── test_span_attention_layer.py ├── test_wiki_reader.py └── test_wordnet.py └── training_config ├── downstream ├── entity_typing.jsonnet ├── semeval2010_task8.jsonnet ├── tacred.jsonnet └── wic.jsonnet └── pretraining ├── knowbert_wiki.jsonnet ├── knowbert_wiki_linker.jsonnet ├── knowbert_wordnet.jsonnet ├── knowbert_wordnet_linker.jsonnet ├── knowbert_wordnet_wiki.jsonnet ├── knowbert_wordnet_wiki_linker.jsonnet └── wordnet_tucker.json /.gitignore: -------------------------------------------------------------------------------- 1 | .pytest_cache/ 2 | .vscode/ 3 | log/ 4 | *.pyc 5 | build 6 | kb.egg-info 7 | dist 8 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/README.md -------------------------------------------------------------------------------- /bin/cloze_task_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/cloze_task_example.py -------------------------------------------------------------------------------- /bin/combine_wordnet_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/combine_wordnet_embeddings.py -------------------------------------------------------------------------------- /bin/create_pretraining_data_for_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/create_pretraining_data_for_bert.py -------------------------------------------------------------------------------- /bin/evaluate_mrr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/evaluate_mrr.py -------------------------------------------------------------------------------- /bin/evaluate_perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/evaluate_perplexity.py -------------------------------------------------------------------------------- /bin/evaluate_wiki_linking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/evaluate_wiki_linking.py -------------------------------------------------------------------------------- /bin/evaluate_wsd_official.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/evaluate_wsd_official.py -------------------------------------------------------------------------------- /bin/extract_wordnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/extract_wordnet.py -------------------------------------------------------------------------------- /bin/preprocess_wsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/preprocess_wsd.py -------------------------------------------------------------------------------- /bin/run_hyperparameter_seeds.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/run_hyperparameter_seeds.sh -------------------------------------------------------------------------------- /bin/semeval2010_task8_scorer-v1.2.pl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/semeval2010_task8_scorer-v1.2.pl -------------------------------------------------------------------------------- /bin/tacred_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/tacred_scorer.py -------------------------------------------------------------------------------- /bin/write_semeval2010_task8_for_official_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/write_semeval2010_task8_for_official_eval.py -------------------------------------------------------------------------------- /bin/write_tacred_for_official_scorer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/write_tacred_for_official_scorer.py -------------------------------------------------------------------------------- /bin/write_wic_for_codalab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/bin/write_wic_for_codalab.py -------------------------------------------------------------------------------- /kb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kb/bert_pretraining_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/bert_pretraining_reader.py -------------------------------------------------------------------------------- /kb/bert_tokenizer_and_candidate_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/bert_tokenizer_and_candidate_generator.py -------------------------------------------------------------------------------- /kb/bert_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/bert_utils.py -------------------------------------------------------------------------------- /kb/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/common.py -------------------------------------------------------------------------------- /kb/dict_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/dict_field.py -------------------------------------------------------------------------------- /kb/entity_linking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/entity_linking.py -------------------------------------------------------------------------------- /kb/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kb/evaluation/classification_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/classification_model.py -------------------------------------------------------------------------------- /kb/evaluation/exponential_average_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/exponential_average_metric.py -------------------------------------------------------------------------------- /kb/evaluation/fbeta_measure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/fbeta_measure.py -------------------------------------------------------------------------------- /kb/evaluation/semeval2010_task8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/semeval2010_task8.py -------------------------------------------------------------------------------- /kb/evaluation/tacred_dataset_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/tacred_dataset_reader.py -------------------------------------------------------------------------------- /kb/evaluation/tacred_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/tacred_predictor.py -------------------------------------------------------------------------------- /kb/evaluation/ultra_fine_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/ultra_fine_reader.py -------------------------------------------------------------------------------- /kb/evaluation/weighted_average.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/weighted_average.py -------------------------------------------------------------------------------- /kb/evaluation/wic_dataset_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/evaluation/wic_dataset_reader.py -------------------------------------------------------------------------------- /kb/include_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/include_all.py -------------------------------------------------------------------------------- /kb/kg_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/kg_embedding.py -------------------------------------------------------------------------------- /kb/kg_probe_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/kg_probe_reader.py -------------------------------------------------------------------------------- /kb/knowbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/knowbert.py -------------------------------------------------------------------------------- /kb/knowbert_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/knowbert_utils.py -------------------------------------------------------------------------------- /kb/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/metrics.py -------------------------------------------------------------------------------- /kb/multitask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/multitask.py -------------------------------------------------------------------------------- /kb/self_attn_bucket_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/self_attn_bucket_iterator.py -------------------------------------------------------------------------------- /kb/span_attention_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/span_attention_layer.py -------------------------------------------------------------------------------- /kb/testing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/testing.py -------------------------------------------------------------------------------- /kb/wiki_linking_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/wiki_linking_reader.py -------------------------------------------------------------------------------- /kb/wiki_linking_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/wiki_linking_util.py -------------------------------------------------------------------------------- /kb/wordnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/kb/wordnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evaluation/test_semeval2010_task8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/evaluation/test_semeval2010_task8.py -------------------------------------------------------------------------------- /tests/evaluation/test_simple_classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/evaluation/test_simple_classifier.py -------------------------------------------------------------------------------- /tests/evaluation/test_tacred_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/evaluation/test_tacred_reader.py -------------------------------------------------------------------------------- /tests/evaluation/test_ultra_fine_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/evaluation/test_ultra_fine_reader.py -------------------------------------------------------------------------------- /tests/evaluation/test_wic_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/evaluation/test_wic_reader.py -------------------------------------------------------------------------------- /tests/fixtures/bert/bert_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/bert/bert_config.json -------------------------------------------------------------------------------- /tests/fixtures/bert/bert_test_fixture.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/bert/bert_test_fixture.tar.gz -------------------------------------------------------------------------------- /tests/fixtures/bert/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/bert/vocab.txt -------------------------------------------------------------------------------- /tests/fixtures/bert/vocab_dir_with_entities_for_tokenizer_and_generator/entity.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/bert/vocab_dir_with_entities_for_tokenizer_and_generator/entity.txt -------------------------------------------------------------------------------- /tests/fixtures/bert/vocab_dir_with_entities_for_tokenizer_and_generator/non_padded_namespaces.txt: -------------------------------------------------------------------------------- 1 | *labels 2 | *tags 3 | -------------------------------------------------------------------------------- /tests/fixtures/bert/vocab_dir_with_entities_for_tokenizer_and_generator/tokens.txt: -------------------------------------------------------------------------------- 1 | @@UNKNOWN@@ 2 | bob 3 | scored 4 | a 5 | hat 6 | trick 7 | the 8 | -------------------------------------------------------------------------------- /tests/fixtures/bert_pretraining/shard1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/bert_pretraining/shard1.txt -------------------------------------------------------------------------------- /tests/fixtures/evaluation/semeval2010_task8/semeval2010_task8.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/evaluation/semeval2010_task8/semeval2010_task8.json -------------------------------------------------------------------------------- /tests/fixtures/evaluation/semeval2010_task8/vocab_entity_markers.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/evaluation/semeval2010_task8/vocab_entity_markers.txt -------------------------------------------------------------------------------- /tests/fixtures/evaluation/ultra_fine/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/evaluation/ultra_fine/train.json -------------------------------------------------------------------------------- /tests/fixtures/evaluation/ultra_fine/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/evaluation/ultra_fine/vocab.txt -------------------------------------------------------------------------------- /tests/fixtures/evaluation/wic/train.data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/evaluation/wic/train.data.txt -------------------------------------------------------------------------------- /tests/fixtures/evaluation/wic/train.gold.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/evaluation/wic/train.gold.txt -------------------------------------------------------------------------------- /tests/fixtures/evaluation/wic/vocab_entity_markers.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/evaluation/wic/vocab_entity_markers.txt -------------------------------------------------------------------------------- /tests/fixtures/kg_embeddings/tucker_wordnet/model.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/kg_embeddings/tucker_wordnet/model.tar.gz -------------------------------------------------------------------------------- /tests/fixtures/kg_embeddings/tucker_wordnet/vocabulary/entity.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/kg_embeddings/tucker_wordnet/vocabulary/entity.txt -------------------------------------------------------------------------------- /tests/fixtures/kg_embeddings/tucker_wordnet/vocabulary/non_padded_namespaces.txt: -------------------------------------------------------------------------------- 1 | *labels 2 | *tags 3 | -------------------------------------------------------------------------------- /tests/fixtures/kg_embeddings/tucker_wordnet/vocabulary/relation.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/kg_embeddings/tucker_wordnet/vocabulary/relation.txt -------------------------------------------------------------------------------- /tests/fixtures/kg_embeddings/wn18rr_dev.txt: -------------------------------------------------------------------------------- 1 | 1 _hypernym 5 2 | -------------------------------------------------------------------------------- /tests/fixtures/kg_embeddings/wn18rr_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/kg_embeddings/wn18rr_train.txt -------------------------------------------------------------------------------- /tests/fixtures/kg_probe/file1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/kg_probe/file1.txt -------------------------------------------------------------------------------- /tests/fixtures/linking/aida.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/linking/aida.txt -------------------------------------------------------------------------------- /tests/fixtures/linking/entities_universe.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/linking/entities_universe.txt -------------------------------------------------------------------------------- /tests/fixtures/linking/priors.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/linking/priors.txt -------------------------------------------------------------------------------- /tests/fixtures/multitask/ccgbank.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/multitask/ccgbank.txt -------------------------------------------------------------------------------- /tests/fixtures/multitask/conll2003.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/multitask/conll2003.txt -------------------------------------------------------------------------------- /tests/fixtures/tacred/LDC2018T24.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/tacred/LDC2018T24.json -------------------------------------------------------------------------------- /tests/fixtures/tacred/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/tacred/vocab.txt -------------------------------------------------------------------------------- /tests/fixtures/wordnet/cat_hat_mask_null_embedding.hdf5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet/cat_hat_mask_null_embedding.hdf5 -------------------------------------------------------------------------------- /tests/fixtures/wordnet/cat_hat_synset_mask_null_vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet/cat_hat_synset_mask_null_vocab.txt -------------------------------------------------------------------------------- /tests/fixtures/wordnet/cat_hat_vocabdir/entity.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet/cat_hat_vocabdir/entity.txt -------------------------------------------------------------------------------- /tests/fixtures/wordnet/cat_hat_vocabdir/non_padded_namespaces.txt: -------------------------------------------------------------------------------- 1 | *labels 2 | *tags 3 | -------------------------------------------------------------------------------- /tests/fixtures/wordnet/cat_hat_vocabdir/tokens.txt: -------------------------------------------------------------------------------- 1 | @@UNKNOWN@@ 2 | bob 3 | scored 4 | a 5 | hat 6 | trick 7 | the 8 | -------------------------------------------------------------------------------- /tests/fixtures/wordnet/entities_cat_hat.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet/entities_cat_hat.jsonl -------------------------------------------------------------------------------- /tests/fixtures/wordnet/entities_fixture.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet/entities_fixture.jsonl -------------------------------------------------------------------------------- /tests/fixtures/wordnet/wsd_dataset.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet/wsd_dataset.json -------------------------------------------------------------------------------- /tests/fixtures/wordnet_wiki_vocab/entity_wiki.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet_wiki_vocab/entity_wiki.txt -------------------------------------------------------------------------------- /tests/fixtures/wordnet_wiki_vocab/entity_wordnet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/fixtures/wordnet_wiki_vocab/entity_wordnet.txt -------------------------------------------------------------------------------- /tests/fixtures/wordnet_wiki_vocab/non_padded_namespaces.txt: -------------------------------------------------------------------------------- 1 | *labels 2 | *tags 3 | -------------------------------------------------------------------------------- /tests/test_bert_pretraining_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_bert_pretraining_reader.py -------------------------------------------------------------------------------- /tests/test_bert_tokenizer_and_candidate_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_bert_tokenizer_and_candidate_generator.py -------------------------------------------------------------------------------- /tests/test_common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_common.py -------------------------------------------------------------------------------- /tests/test_dict_field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_dict_field.py -------------------------------------------------------------------------------- /tests/test_entity_linking.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_entity_linking.py -------------------------------------------------------------------------------- /tests/test_kg_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_kg_embedding.py -------------------------------------------------------------------------------- /tests/test_kg_probe_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_kg_probe_reader.py -------------------------------------------------------------------------------- /tests/test_knowbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_knowbert.py -------------------------------------------------------------------------------- /tests/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_metrics.py -------------------------------------------------------------------------------- /tests/test_multitask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_multitask.py -------------------------------------------------------------------------------- /tests/test_self_attn_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_self_attn_iterator.py -------------------------------------------------------------------------------- /tests/test_span_attention_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_span_attention_layer.py -------------------------------------------------------------------------------- /tests/test_wiki_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_wiki_reader.py -------------------------------------------------------------------------------- /tests/test_wordnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/tests/test_wordnet.py -------------------------------------------------------------------------------- /training_config/downstream/entity_typing.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/downstream/entity_typing.jsonnet -------------------------------------------------------------------------------- /training_config/downstream/semeval2010_task8.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/downstream/semeval2010_task8.jsonnet -------------------------------------------------------------------------------- /training_config/downstream/tacred.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/downstream/tacred.jsonnet -------------------------------------------------------------------------------- /training_config/downstream/wic.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/downstream/wic.jsonnet -------------------------------------------------------------------------------- /training_config/pretraining/knowbert_wiki.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/pretraining/knowbert_wiki.jsonnet -------------------------------------------------------------------------------- /training_config/pretraining/knowbert_wiki_linker.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/pretraining/knowbert_wiki_linker.jsonnet -------------------------------------------------------------------------------- /training_config/pretraining/knowbert_wordnet.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/pretraining/knowbert_wordnet.jsonnet -------------------------------------------------------------------------------- /training_config/pretraining/knowbert_wordnet_linker.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/pretraining/knowbert_wordnet_linker.jsonnet -------------------------------------------------------------------------------- /training_config/pretraining/knowbert_wordnet_wiki.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/pretraining/knowbert_wordnet_wiki.jsonnet -------------------------------------------------------------------------------- /training_config/pretraining/knowbert_wordnet_wiki_linker.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/pretraining/knowbert_wordnet_wiki_linker.jsonnet -------------------------------------------------------------------------------- /training_config/pretraining/wordnet_tucker.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/allenai/kb/HEAD/training_config/pretraining/wordnet_tucker.json --------------------------------------------------------------------------------