diff --git a/ConvLab-3 b/ConvLab-3 new file mode 160000 index 0000000..60f4e56 --- /dev/null +++ b/ConvLab-3 @@ -0,0 +1 @@ +Subproject commit 60f4e5641f93e99b8d61b49cf5fd6dc818a83c4c diff --git a/MachineLearningNLG.py b/MachineLearningNLG.py index f9434dd..39d9be0 100644 --- a/MachineLearningNLG.py +++ b/MachineLearningNLG.py @@ -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)) diff --git a/train_nlg.py b/train_nlg.py index 032083c..6644e4e 100644 --- a/train_nlg.py +++ b/train_nlg.py @@ -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