├── .dockerignore ├── .gitignore ├── .pylintrc ├── .travis.yml ├── Dockerfile ├── README.md ├── codecov.yml ├── experiments ├── alias_copynet.jsonnet ├── awd-lstm-lm-modified.jsonnet ├── awd-lstm-lm-simple.jsonnet ├── awd-lstm-lm.jsonnet ├── entity_disc.jsonnet ├── entity_disc_conll2012_test.jsonnet ├── entity_disc_test.jsonnet ├── entity_nlm.jsonnet ├── entity_nlm_conll2012_test.jsonnet ├── entity_nlm_test.jsonnet ├── kglm-disc.jsonnet ├── kglm.jsonnet ├── kglm_test.jsonnet └── simplified.jsonl ├── kglm ├── __init__.py ├── commands │ ├── __init__.py │ ├── complete_the_sentence.py │ ├── evaluate_perplexity.py │ └── sample.py ├── common │ ├── __init__.py │ ├── testing │ │ ├── __init__.py │ │ └── kglm_model_test_case.py │ └── typing.py ├── data │ ├── __init__.py │ ├── alias_database.py │ ├── dataset_readers │ │ ├── __init__.py │ │ ├── conll2012.py │ │ └── enhanced_wikitext.py │ ├── extended_vocabulary.py │ ├── fields │ │ ├── __init__.py │ │ ├── global_object.py │ │ └── sequential_array.py │ └── iterators │ │ ├── __init__.py │ │ ├── awd_iterator.py │ │ ├── fancy_iterator.py │ │ └── split_iterator.py ├── models │ ├── __init__.py │ ├── alias_copynet.py │ ├── awd_lstm.py │ ├── entity_disc.py │ ├── entity_nlm.py │ ├── kglm.py │ ├── kglm_disc.py │ ├── no_story.py │ └── simplified.py ├── modules │ ├── __init__.py │ ├── dynamic_embeddings.py │ ├── embed_regularize.py │ ├── knowledge_graph_lookup.py │ ├── locked_dropout.py │ ├── recent_entities.py │ ├── splitcross.py │ └── weight_drop.py ├── nn │ ├── __init__.py │ └── util.py ├── predictors │ ├── __init__.py │ ├── complete_the_sentence.py │ └── kglm.py ├── run.py ├── tests │ ├── __init__.py │ ├── data │ │ ├── __init__.py │ │ ├── alias_database_test.py │ │ ├── fancy_iterator_test.py │ │ └── vocabulary_test.py │ ├── dataset_readers │ │ ├── __init__.py │ │ ├── conll2012_test.py │ │ └── enhanced_wikitext_test.py │ ├── fixtures │ │ ├── cloze.jsonl │ │ ├── coref.gold_conll │ │ ├── enhanced-wikitext-test │ │ │ ├── alias.pkl │ │ │ ├── knowledge_graph.pkl │ │ │ ├── train.jsonl │ │ │ └── vocab │ │ │ │ ├── entity_ids.txt │ │ │ │ ├── non_padded_namespaces.txt │ │ │ │ ├── raw_entity_ids.txt │ │ │ │ ├── relations.txt │ │ │ │ ├── tokens.txt │ │ │ │ └── tokens_unk.txt │ │ ├── kglm.model.tar.gz │ │ ├── mini.alias.pkl │ │ ├── mini.relation.pkl │ │ └── training_config │ │ │ ├── alias_copynet.json │ │ │ ├── awd-lstm-lm.json │ │ │ ├── entity_nlm.json │ │ │ ├── kglm-disc.json │ │ │ ├── kglm-disc.no-shortlist.json │ │ │ ├── kglm.json │ │ │ ├── kglm.no-shortlist.json │ │ │ └── no-story.json │ ├── models │ │ ├── __init__.py │ │ ├── awd_lstm_test.py │ │ ├── entity_nlm_test.py │ │ ├── kglm_test.py │ │ └── no_story_test.py │ ├── modules │ │ ├── test_dynamic_entity_embeddings.py │ │ ├── test_knowledge_graph_lookup.py │ │ ├── test_locked_dropout.py │ │ └── test_recent_entities.py │ ├── predictors │ │ └── complete_the_sentence_test.py │ └── training │ │ └── nt_asgd_test.py └── training │ ├── __init__.py │ ├── metrics │ ├── __init__.py │ └── perplexity.py │ ├── nt_asgd.py │ └── trainer.py ├── pytest.ini └── requirements.txt /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/.dockerignore -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/.gitignore -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/.pylintrc -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/.travis.yml -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/Dockerfile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/README.md -------------------------------------------------------------------------------- /codecov.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/codecov.yml -------------------------------------------------------------------------------- /experiments/alias_copynet.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/alias_copynet.jsonnet -------------------------------------------------------------------------------- /experiments/awd-lstm-lm-modified.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/awd-lstm-lm-modified.jsonnet -------------------------------------------------------------------------------- /experiments/awd-lstm-lm-simple.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/awd-lstm-lm-simple.jsonnet -------------------------------------------------------------------------------- /experiments/awd-lstm-lm.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/awd-lstm-lm.jsonnet -------------------------------------------------------------------------------- /experiments/entity_disc.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/entity_disc.jsonnet -------------------------------------------------------------------------------- /experiments/entity_disc_conll2012_test.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/entity_disc_conll2012_test.jsonnet -------------------------------------------------------------------------------- /experiments/entity_disc_test.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/entity_disc_test.jsonnet -------------------------------------------------------------------------------- /experiments/entity_nlm.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/entity_nlm.jsonnet -------------------------------------------------------------------------------- /experiments/entity_nlm_conll2012_test.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/entity_nlm_conll2012_test.jsonnet -------------------------------------------------------------------------------- /experiments/entity_nlm_test.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/entity_nlm_test.jsonnet -------------------------------------------------------------------------------- /experiments/kglm-disc.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/kglm-disc.jsonnet -------------------------------------------------------------------------------- /experiments/kglm.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/kglm.jsonnet -------------------------------------------------------------------------------- /experiments/kglm_test.jsonnet: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/kglm_test.jsonnet -------------------------------------------------------------------------------- /experiments/simplified.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/experiments/simplified.jsonl -------------------------------------------------------------------------------- /kglm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/__init__.py -------------------------------------------------------------------------------- /kglm/commands/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/commands/__init__.py -------------------------------------------------------------------------------- /kglm/commands/complete_the_sentence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/commands/complete_the_sentence.py -------------------------------------------------------------------------------- /kglm/commands/evaluate_perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/commands/evaluate_perplexity.py -------------------------------------------------------------------------------- /kglm/commands/sample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/commands/sample.py -------------------------------------------------------------------------------- /kglm/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kglm/common/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/common/testing/__init__.py -------------------------------------------------------------------------------- /kglm/common/testing/kglm_model_test_case.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/common/testing/kglm_model_test_case.py -------------------------------------------------------------------------------- /kglm/common/typing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/common/typing.py -------------------------------------------------------------------------------- /kglm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/__init__.py -------------------------------------------------------------------------------- /kglm/data/alias_database.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/alias_database.py -------------------------------------------------------------------------------- /kglm/data/dataset_readers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/dataset_readers/__init__.py -------------------------------------------------------------------------------- /kglm/data/dataset_readers/conll2012.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/dataset_readers/conll2012.py -------------------------------------------------------------------------------- /kglm/data/dataset_readers/enhanced_wikitext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/dataset_readers/enhanced_wikitext.py -------------------------------------------------------------------------------- /kglm/data/extended_vocabulary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/extended_vocabulary.py -------------------------------------------------------------------------------- /kglm/data/fields/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/fields/__init__.py -------------------------------------------------------------------------------- /kglm/data/fields/global_object.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/fields/global_object.py -------------------------------------------------------------------------------- /kglm/data/fields/sequential_array.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/fields/sequential_array.py -------------------------------------------------------------------------------- /kglm/data/iterators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/iterators/__init__.py -------------------------------------------------------------------------------- /kglm/data/iterators/awd_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/iterators/awd_iterator.py -------------------------------------------------------------------------------- /kglm/data/iterators/fancy_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/iterators/fancy_iterator.py -------------------------------------------------------------------------------- /kglm/data/iterators/split_iterator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/data/iterators/split_iterator.py -------------------------------------------------------------------------------- /kglm/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kglm/models/alias_copynet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/alias_copynet.py -------------------------------------------------------------------------------- /kglm/models/awd_lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/awd_lstm.py -------------------------------------------------------------------------------- /kglm/models/entity_disc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/entity_disc.py -------------------------------------------------------------------------------- /kglm/models/entity_nlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/entity_nlm.py -------------------------------------------------------------------------------- /kglm/models/kglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/kglm.py -------------------------------------------------------------------------------- /kglm/models/kglm_disc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/kglm_disc.py -------------------------------------------------------------------------------- /kglm/models/no_story.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/no_story.py -------------------------------------------------------------------------------- /kglm/models/simplified.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/models/simplified.py -------------------------------------------------------------------------------- /kglm/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/__init__.py -------------------------------------------------------------------------------- /kglm/modules/dynamic_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/dynamic_embeddings.py -------------------------------------------------------------------------------- /kglm/modules/embed_regularize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/embed_regularize.py -------------------------------------------------------------------------------- /kglm/modules/knowledge_graph_lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/knowledge_graph_lookup.py -------------------------------------------------------------------------------- /kglm/modules/locked_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/locked_dropout.py -------------------------------------------------------------------------------- /kglm/modules/recent_entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/recent_entities.py -------------------------------------------------------------------------------- /kglm/modules/splitcross.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/splitcross.py -------------------------------------------------------------------------------- /kglm/modules/weight_drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/modules/weight_drop.py -------------------------------------------------------------------------------- /kglm/nn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kglm/nn/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/nn/util.py -------------------------------------------------------------------------------- /kglm/predictors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/predictors/__init__.py -------------------------------------------------------------------------------- /kglm/predictors/complete_the_sentence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/predictors/complete_the_sentence.py -------------------------------------------------------------------------------- /kglm/predictors/kglm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/predictors/kglm.py -------------------------------------------------------------------------------- /kglm/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/run.py -------------------------------------------------------------------------------- /kglm/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kglm/tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kglm/tests/data/alias_database_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/data/alias_database_test.py -------------------------------------------------------------------------------- /kglm/tests/data/fancy_iterator_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/data/fancy_iterator_test.py -------------------------------------------------------------------------------- /kglm/tests/data/vocabulary_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/data/vocabulary_test.py -------------------------------------------------------------------------------- /kglm/tests/dataset_readers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kglm/tests/dataset_readers/conll2012_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/dataset_readers/conll2012_test.py -------------------------------------------------------------------------------- /kglm/tests/dataset_readers/enhanced_wikitext_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/dataset_readers/enhanced_wikitext_test.py -------------------------------------------------------------------------------- /kglm/tests/fixtures/cloze.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/cloze.jsonl -------------------------------------------------------------------------------- /kglm/tests/fixtures/coref.gold_conll: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/coref.gold_conll -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/alias.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/alias.pkl -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/knowledge_graph.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/knowledge_graph.pkl -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/train.jsonl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/train.jsonl -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/vocab/entity_ids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/vocab/entity_ids.txt -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/vocab/non_padded_namespaces.txt: -------------------------------------------------------------------------------- 1 | *tags 2 | *labels 3 | *unk 4 | -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/vocab/raw_entity_ids.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/vocab/raw_entity_ids.txt -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/vocab/relations.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/vocab/relations.txt -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/vocab/tokens.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/vocab/tokens.txt -------------------------------------------------------------------------------- /kglm/tests/fixtures/enhanced-wikitext-test/vocab/tokens_unk.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/enhanced-wikitext-test/vocab/tokens_unk.txt -------------------------------------------------------------------------------- /kglm/tests/fixtures/kglm.model.tar.gz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/kglm.model.tar.gz -------------------------------------------------------------------------------- /kglm/tests/fixtures/mini.alias.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/mini.alias.pkl -------------------------------------------------------------------------------- /kglm/tests/fixtures/mini.relation.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/mini.relation.pkl -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/alias_copynet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/alias_copynet.json -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/awd-lstm-lm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/awd-lstm-lm.json -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/entity_nlm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/entity_nlm.json -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/kglm-disc.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/kglm-disc.json -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/kglm-disc.no-shortlist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/kglm-disc.no-shortlist.json -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/kglm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/kglm.json -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/kglm.no-shortlist.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/kglm.no-shortlist.json -------------------------------------------------------------------------------- /kglm/tests/fixtures/training_config/no-story.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/fixtures/training_config/no-story.json -------------------------------------------------------------------------------- /kglm/tests/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /kglm/tests/models/awd_lstm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/models/awd_lstm_test.py -------------------------------------------------------------------------------- /kglm/tests/models/entity_nlm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/models/entity_nlm_test.py -------------------------------------------------------------------------------- /kglm/tests/models/kglm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/models/kglm_test.py -------------------------------------------------------------------------------- /kglm/tests/models/no_story_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/models/no_story_test.py -------------------------------------------------------------------------------- /kglm/tests/modules/test_dynamic_entity_embeddings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/modules/test_dynamic_entity_embeddings.py -------------------------------------------------------------------------------- /kglm/tests/modules/test_knowledge_graph_lookup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/modules/test_knowledge_graph_lookup.py -------------------------------------------------------------------------------- /kglm/tests/modules/test_locked_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/modules/test_locked_dropout.py -------------------------------------------------------------------------------- /kglm/tests/modules/test_recent_entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/modules/test_recent_entities.py -------------------------------------------------------------------------------- /kglm/tests/predictors/complete_the_sentence_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/predictors/complete_the_sentence_test.py -------------------------------------------------------------------------------- /kglm/tests/training/nt_asgd_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/tests/training/nt_asgd_test.py -------------------------------------------------------------------------------- /kglm/training/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/training/__init__.py -------------------------------------------------------------------------------- /kglm/training/metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/training/metrics/__init__.py -------------------------------------------------------------------------------- /kglm/training/metrics/perplexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/training/metrics/perplexity.py -------------------------------------------------------------------------------- /kglm/training/nt_asgd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/training/nt_asgd.py -------------------------------------------------------------------------------- /kglm/training/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/kglm/training/trainer.py -------------------------------------------------------------------------------- /pytest.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/pytest.ini -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/rloganiv/kglm-model/HEAD/requirements.txt --------------------------------------------------------------------------------