def _get_attack_classifier_list(**kwargs): if framework == "pytorch": return get_attack_classifier_pt(**kwargs) raise ARTTestFixtureNotImplemented( "no estimator available", image_dl_estimator_for_attack.__name__, framework)
def _get_attack_classifier_list(one_classifier=False, **kwargs): if framework == "keras": logging.warning( "{0} doesn't have an image attack defined yet".format( framework)) classifier_list = None if framework == "tensorflow": logging.warning( "{0} doesn't have an image attack defined yet".format( framework)) classifier_list = None if framework == "pytorch": classifier_list = [get_attack_classifier_pt(**kwargs)] if framework == "scikitlearn": logging.warning( "{0} doesn't have an image attack defined yet".format( framework)) classifier_list = None if classifier_list is None: return None if one_classifier: return classifier_list[0] return classifier_list