test- commit
This commit is contained in:
parent
0b2efaa3f4
commit
7306fdb83a
1
ConvLab-3
Submodule
1
ConvLab-3
Submodule
@ -0,0 +1 @@
|
||||
Subproject commit 60f4e5641f93e99b8d61b49cf5fd6dc818a83c4c
|
@ -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))
|
||||
|
@ -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
|
||||
|
Loading…
Reference in New Issue
Block a user