diff --git a/train.py b/train.py index 2c502001a16b1dbb13f19082f4c39ecc4dfbb4c5..8505431ecabdd47b68b8cf232f58f92ea5697ea4 100644 --- a/train.py +++ b/train.py @@ -9,7 +9,7 @@ batch_size = int(Configuration.modelTrainingConfig['batch_size']) nb_sentences = batch_size * 200 epochs = int(Configuration.modelTrainingConfig['epoch']) -file_path_axiom_links = 'Datasets/aa1_links_dataset_links.csv' +file_path_axiom_links = 'Datasets/goldANDsilver_dataset_links.csv' df_axiom_links = read_csv_pgbar(file_path_axiom_links, nb_sentences) sentences_batch = df_axiom_links["Sentences"].tolist()