First model version

test.py

@@ -0,0 +1,30 @@
+import torch
+from textattack.models.helpers import LSTMForClassification
+from textattack.models.wrappers import PyTorchModelWrapper
+
+model = LSTMForClassification.from_pretrained('/root/attack/outputs/2024-02-23-10-43-21-925203/best_model')
+model = PyTorchModelWrapper(model, model.tokenizer)
+
+from textattack.transformations import WordSwapMaskedLM
+from textattack.search_methods import GreedyWordSwapWIR
+from textattack.datasets import Dataset
+
+transformation = WordSwapMaskedLM()
+search_method = GreedyWordSwapWIR('unk')
+
+from textattack import Attack, AttackArgs, Attacker
+from textattack.goal_functions import UntargetedClassification
+
+dataset = Dataset([('I like play basketball.', 1)])
+
+goal_function = UntargetedClassification(model)
+
+attack_args = AttackArgs(query_budget=10000)
+
+attack = Attack(goal_function, [], transformation, search_method)
+attacker = Attacker(attack, dataset)
+attacker.attack_dataset()
+
+
+
+