From e5c6a8a8182b885f8333f45bd01678ed1dd83b61 Mon Sep 17 00:00:00 2001 From: Caroline DE POURTALES <cdepourt@montana.irit.fr> Date: Wed, 18 May 2022 17:02:34 +0200 Subject: [PATCH] update utils --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 2c50200..8505431 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() -- GitLab