OpenAttack

1
import sys, os
2
sys.path.insert(0, os.path.join(
3
    os.path.dirname(os.path.abspath(__file__)),
4
    ".."
5
))
6

7
import OpenAttack
8
def get_attackers(dataset, clsf):
9

10
    # rules = OpenAttack.attackers.SEAAttacker.get_rules(clsf, dataset)
11

12
    triggers = OpenAttack.attackers.UATAttacker.get_triggers(clsf, dataset, clsf.tokenizer)
13
    print(triggers)
14

15
    attackers = [
16
        OpenAttack.attackers.FDAttacker(token_unk=clsf.token_unk),
17
        OpenAttack.attackers.UATAttacker(triggers=triggers),
18
        OpenAttack.attackers.TextBuggerAttacker(),
19
        OpenAttack.attackers.TextFoolerAttacker(),
20
        OpenAttack.attackers.VIPERAttacker(),
21
        OpenAttack.attackers.DeepWordBugAttacker(),
22
        OpenAttack.attackers.GANAttacker(),
23
        OpenAttack.attackers.GeneticAttacker(),
24
        OpenAttack.attackers.HotFlipAttacker(),
25
        OpenAttack.attackers.PWWSAttacker(),
26
        OpenAttack.attackers.SCPNAttacker(),
27
        OpenAttack.attackers.PSOAttacker(),
28
        OpenAttack.attackers.BAEAttacker(),
29
        OpenAttack.attackers.BERTAttacker()
30
    ]
31
    
32
    return attackers
33