Skip to content
Snippets Groups Projects
Commit e5c6a8a8 authored by Caroline DE POURTALES's avatar Caroline DE POURTALES
Browse files

update utils

parent b5a9eeab
Branches
No related tags found
2 merge requests!6Linker with transformer,!5Linker with transformer
Hide whitespace changes
Inline Side-by-side
Showing
with 1 addition and 1 deletion
train.py
+ 1
− 1
View file @ e5c6a8a8
......@@ -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()
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment