s495719/GOATS
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

test- commit

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
ConvLab-3 Submodule

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

5
MachineLearningNLG.py
View File

@ -33,15 +33,14 @@ class MachineLearningNLG:
return translated_response return translated_response
def generate(self, action): def generate(self, action):
# Przyjmujemy, że 'action' jest formatowanym stringiem, który jest przekazywany do self.nlg
return self.nlg(action) return self.nlg(action)
def init_session(self): def init_session(self):
pass # Dodanie pustej metody init_session pass
# Przykład użycia # Przykład użycia
if __name__ == "__main__": if __name__ == "__main__":
nlg = MachineLearningNLG() 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)) print(nlg.nlg(system_act))

2
train_nlg.py
View File

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