OpenAttack

1
'''
2
This example code shows how to conduct adversarial attacks against a Chinese review classification model using PWWS
3
'''
4
import OpenAttack
5
import datasets
6

7
def dataset_mapping(x):
8
    return {
9
        "x": x["review_body"],
10
        "y": x["stars"],
11
    }
12

13
def main():
14
    print("New Attacker")
15
    attacker = OpenAttack.attackers.PWWSAttacker(lang="chinese")
16

17
    print("Building model")
18
    victim = OpenAttack.loadVictim("BERT.AMAZON_ZH").to("cuda:0")
19

20
    print("Loading dataset")
21
    dataset = datasets.load_dataset("amazon_reviews_multi",'zh',split="train[:20]").map(function=dataset_mapping)
22

23
    print("Start attack")
24
    attack_eval = OpenAttack.AttackEval(attacker, victim)
25
    attack_eval.eval(dataset, visualize=True, progress_bar=True)
26

27
if __name__ == "__main__":
28
    main()
29