import BERT_eval_sim_definition_lemma.test as task1
import BERT_eval_sister_terms_similarity.test as task2
import BERT_eval_definitions_sister_terms_similarity.test as task3

model_name = 'bert-base-uncased'

base_data_dir = 'data_BERT'

print("DEFINITIONS VS EMBEDDINGS")
output_path = 'data_BERT/task1_sim_definition_lemma'
task1.evaluate_definition_lemma_embeddings(model_name=model_name, base_data_dir=base_data_dir,
                                           output_path=output_path)

sister_terms_output_path = 'data_BERT/task2_sister_terms_similarity'
task2.evaluate_sister_terms_similarity(model_name=model_name, base_data_dir=base_data_dir,
                                       sister_terms_path=sister_terms_output_path)

oov_sister_terms_output_path = 'data_BERT/task3_definitions_and_sister_similarity'
task3.evaluate_sister_terms_definitions_similarity(model_name=model_name, base_data_dir=base_data_dir,
                                                   oov_sister_terms_path=oov_sister_terms_output_path)