From 9ab62795cdb3990b3994033ff60eb5e9d08394a6 Mon Sep 17 00:00:00 2001 From: "ErnestsLavrinovicsalias]" Date: Wed, 18 Jan 2023 18:01:38 +0100 Subject: [PATCH] Add hyperparameter control to set_triggers function --- OpenAttack/attackers/uat/__init__.py | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) diff --git a/OpenAttack/attackers/uat/__init__.py b/OpenAttack/attackers/uat/__init__.py index 72a4a89..d44a96d 100644 --- a/OpenAttack/attackers/uat/__init__.py +++ b/OpenAttack/attackers/uat/__init__.py @@ -53,8 +53,22 @@ def __init__(self, def set_triggers(self, victim : Classifier, - dataset : datasets.Dataset,): - self.triggers = self.get_triggers(victim, dataset, self.tokenizer) + dataset : datasets.Dataset, + epoch : int = 5, + batch_size : int = 5, + trigger_len : int = 3, + beam_size : int = 5, + lang = None): + + self.triggers = self.get_triggers(victim, + dataset, + self.tokenizer, + epoch=epoch, + batch_size=batch_size, + trigger_len=trigger_len, + beam_size=beam_size, + lang=lang + ) def attack(self, victim: Classifier, sentence : str, goal : ClassifierGoal): trigger_sent = self.tokenizer.detokenize( self.triggers + self.tokenizer.tokenize(sentence, pos_tagging=False) )