diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py index 9e6bc8b0a57d661228bfed1e3116229cf2625691..523ec3391d4e781edf5218a4acb621cdd8b3e5e1 100644 --- a/mhcflurry/class1_neural_network.py +++ b/mhcflurry/class1_neural_network.py @@ -81,7 +81,7 @@ class Class1NeuralNetwork(object): validation_split=0.1, early_stopping=True, minibatch_size=128, - random_negative_rate=0.2, + random_negative_rate=0.0, random_negative_constant=25, random_negative_affinity_min=20000.0, random_negative_affinity_max=50000.0, diff --git a/test/test_train_and_related_commands.py b/test/test_train_and_related_commands.py index c2fdd23de4ddf3780c9743f65970abf2509fc50f..0e8e3d7645c1e781963c5357285c80fff747ac48 100644 --- a/test/test_train_and_related_commands.py +++ b/test/test_train_and_related_commands.py @@ -64,7 +64,6 @@ def run_and_check(n_jobs=0): "--allele", "HLA-A*02:01", "HLA-A*03:01", "--out-models-dir", models_dir, "--num-jobs", str(n_jobs), - "--ignore-inequalities", ] print("Running with args: %s" % args) subprocess.check_call(args) @@ -117,7 +116,6 @@ def run_and_check_with_model_selection(n_jobs=1): "--allele", "HLA-A*02:01", "HLA-A*03:01", "--out-models-dir", models_dir1, "--num-jobs", str(n_jobs), - "--ignore-inequalities", "--held-out-fraction-reciprocal", "10", "--n-models", "1", ]