6c7cd14922969eb8aa7f9276e5702e081d13538a,tests/attacks/inference/test_attribute_inference.py,,test_white_box,#Any#Any#,31

Before Change

        test_diff = np.abs(inferred_test - x_test_feature)
        assert sum(train_diff) < sum(test_diff)
        //TODO: to be removed
        break

if __name__ == "__main__":
    pytest.cmdline.main("-q {} --mlFramework=scikitlearn --durations=0".format(__file__).split(" "))

After Change

        // print(classifier)
        attack = AttributeInferenceWhiteBox(classifier, attack_feature=attack_feature)
        x_train_predictions = np.array([np.argmax(arr) for arr in classifier.predict(x_train_iris)]).reshape(-1,1)
        x_test_predictions = np.array([np.argmax(arr) for arr in classifier.predict(x_test_iris)]).reshape(-1,1)
        inferred_train = attack.infer(x_train_for_attack, x_train_predictions, values=values, priors=priors)
        inferred_test = attack.infer(x_test_for_attack, x_test_predictions, values=values, priors=priors)
        train_diff = np.abs(inferred_train - x_train_feature)
        test_diff = np.abs(inferred_test - x_test_feature)

Instances