From e9a9be87953144209014a4769e557e02c72a4679 Mon Sep 17 00:00:00 2001
From: Frederik Arnold <frederik.arnold@hu-berlin.de>
Date: Fri, 26 Apr 2024 07:52:33 +0200
Subject: [PATCH] Update training evaluation
---
indiquo/training/candidate/TrainCandidateClassifierST.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/indiquo/training/candidate/TrainCandidateClassifierST.py b/indiquo/training/candidate/TrainCandidateClassifierST.py
index 878f933..10179d7 100644
--- a/indiquo/training/candidate/TrainCandidateClassifierST.py
+++ b/indiquo/training/candidate/TrainCandidateClassifierST.py
@@ -31,9 +31,9 @@ def train(train_folder_path, output_folder_path, model_name):
train_dataloader = DataLoader(train_examples, shuffle=True, batch_size=16)
train_loss = losses.ContrastiveLoss(model=model)
- evaluator = evaluation.BinaryClassificationEvaluator(val_sentences_1, val_sentences_2, val_labels, write_csv=False)
+ evaluator = evaluation.BinaryClassificationEvaluator(val_sentences_1, val_sentences_2, val_labels)
num_epochs = 5
warmup_steps = math.ceil(len(train_dataloader) * num_epochs * 0.1) # 10% of train data for warm-up
model.fit(train_objectives=[(train_dataloader, train_loss)], epochs=num_epochs, warmup_steps=warmup_steps,
- evaluator=evaluator, evaluation_steps=1000, output_path=output_folder_path)
+ evaluator=evaluator, output_path=output_folder_path)
--
GitLab