Delete test.py
Browse files
test.py
DELETED
@@ -1,30 +0,0 @@
|
|
1 |
-
import torch
|
2 |
-
from textattack.models.helpers import LSTMForClassification
|
3 |
-
from textattack.models.wrappers import PyTorchModelWrapper
|
4 |
-
|
5 |
-
model = LSTMForClassification.from_pretrained('/root/attack/outputs/2024-02-23-10-43-21-925203/best_model')
|
6 |
-
model = PyTorchModelWrapper(model, model.tokenizer)
|
7 |
-
|
8 |
-
from textattack.transformations import WordSwapMaskedLM
|
9 |
-
from textattack.search_methods import GreedyWordSwapWIR
|
10 |
-
from textattack.datasets import Dataset
|
11 |
-
|
12 |
-
transformation = WordSwapMaskedLM()
|
13 |
-
search_method = GreedyWordSwapWIR('unk')
|
14 |
-
|
15 |
-
from textattack import Attack, AttackArgs, Attacker
|
16 |
-
from textattack.goal_functions import UntargetedClassification
|
17 |
-
|
18 |
-
dataset = Dataset([('I like play basketball.', 1)])
|
19 |
-
|
20 |
-
goal_function = UntargetedClassification(model)
|
21 |
-
|
22 |
-
attack_args = AttackArgs(query_budget=10000)
|
23 |
-
|
24 |
-
attack = Attack(goal_function, [], transformation, search_method)
|
25 |
-
attacker = Attacker(attack, dataset)
|
26 |
-
attacker.attack_dataset()
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|