5a8d8e4c155de0b27d3e30bb165f7f375209ecf9,train.py,,,#,15

Before Change

// --------------------------------------------------------------------------------------------- GET CLASSIFIER

// get MNIST
(X_train, Y_train), (X_test, Y_test) = load_mnist()

if os.path.isfile(args.dataset):
    X_train = np.load(args.dataset)
    Y_train = Y_train if "train.npy" in args.dataset else Y_test

After Change

    else:

        if args.defences:
            defences = "-".join(args.defences)
        else:
            defences = ""
        MODEL_PATH = os.path.join(os.path.abspath(DATA_PATH), "classifiers", args.dataset, args.classifier, args.act,
                                  defences)

    v_print("Classifier saved in", MODEL_PATH)

Instances