Commit 974907b4 authored by Nandeesh Patel Gowdru Prabushanker's avatar Nandeesh Patel Gowdru Prabushanker
Browse files

Merge branch 'embedding-index' into 'master'

Instruction to train a single model

See merge request !6
parents 4188dd87 2c45ec03
......@@ -149,6 +149,7 @@ def evaluate(x_test, y_test, doc_emb):
if __name__ == "__main__":
data, embeddings = prepare_data()
# use dataset_names = ["aida_train"] to train and save only one model
dataset_names = ["AQUAINT", "aida_complete", "aida_train", "aida_testa", "aida_testb", "ACE2004", "MSNBC", "iitb",
"N3-Reuters-128", "N3-RSS-500", "oke-challenge-task1-example", "oke-challenge-task1-gs",
"oke-challenge-task1-eval", "KORE50", "spotlight"]
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment