Commit 2c45ec03 authored by Nandeesh Patel Gowdru Prabushanker's avatar Nandeesh Patel Gowdru Prabushanker
Browse files

Instruction to train a single model

parent f33bdf67
......@@ -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