OpenBackdoor

1
from .poisoner import Poisoner
2
from .badnets_poisoner import BadNetsPoisoner
3
from .ep_poisoner import EPPoisoner
4
from .sos_poisoner import SOSPoisoner
5
from .synbkd_poisoner import SynBkdPoisoner
6
from .stylebkd_poisoner import StyleBkdPoisoner
7
from .addsent_poisoner import AddSentPoisoner
8
from .trojanlm_poisoner import TrojanLMPoisoner
9
from .neuba_poisoner import NeuBAPoisoner
10
from .por_poisoner import PORPoisoner
11
from .lwp_poisoner import LWPPoisoner
12

13
POISONERS = {
14
    "base": Poisoner,
15
    "badnets": BadNetsPoisoner,
16
    "ep": EPPoisoner,
17
    "sos": SOSPoisoner,
18
    "synbkd": SynBkdPoisoner,
19
    "stylebkd": StyleBkdPoisoner,
20
    "addsent": AddSentPoisoner,
21
    "trojanlm": TrojanLMPoisoner,
22
    "neuba": NeuBAPoisoner,
23
    "por": PORPoisoner,
24
    "lwp": LWPPoisoner
25
}
26

27
def load_poisoner(config):
28
    return POISONERS[config["name"].lower()](**config)
29