From 30bbc4a8d1f0d9f5625f50aaf4f4604d4220e6ae Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Tue, 20 Feb 2018 14:32:43 -0500
Subject: [PATCH] test tweak
---
mhcflurry/class1_neural_network.py | 2 +-
test/test_train_and_related_commands.py | 2 --
2 files changed, 1 insertion(+), 3 deletions(-)
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 9e6bc8b0..523ec339 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 c2fdd23d..0e8e3d76 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",
]
--
GitLab