test- commit

This commit is contained in:
Maciej Matusz 2024-06-04 10:56:40 +02:00
parent 0b2efaa3f4
commit 7306fdb83a
3 changed files with 4 additions and 4 deletions

1
ConvLab-3 Submodule

@ -0,0 +1 @@
Subproject commit 60f4e5641f93e99b8d61b49cf5fd6dc818a83c4c

View File

@ -33,15 +33,14 @@ class MachineLearningNLG:
return translated_response
def generate(self, action):
# Przyjmujemy, że 'action' jest formatowanym stringiem, który jest przekazywany do self.nlg
return self.nlg(action)
def init_session(self):
pass # Dodanie pustej metody init_session
pass
# Przykład użycia
if __name__ == "__main__":
nlg = MachineLearningNLG()
system_act = "inform(date.from=15.07, date.to=22.07)"
system_act = "inform(people.kids.ages=[4,9])"
print(nlg.nlg(system_act))

View File

@ -49,7 +49,7 @@ training_args = Seq2SeqTrainingArguments(
per_device_eval_batch_size=16,
predict_with_generate=True,
learning_rate=5e-5,
num_train_epochs=3,
num_train_epochs=10,
evaluation_strategy="epoch",
save_strategy="epoch",
save_total_limit=None, # Wyłącz rotację punktów kontrolnych