From a1a2cdb83c927a229faaaf1028d2c3e43f755392 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 18 Jan 2018 08:11:23 -0500
Subject: [PATCH] update tests
---
test/test_class1_affinity_predictor.py | 2 ++
test/test_class1_neural_network.py | 3 ++-
2 files changed, 4 insertions(+), 1 deletion(-)
diff --git a/test/test_class1_affinity_predictor.py b/test/test_class1_affinity_predictor.py
index 320b3961..8532b76a 100644
--- a/test/test_class1_affinity_predictor.py
+++ b/test/test_class1_affinity_predictor.py
@@ -96,6 +96,7 @@ def test_a1_known_epitopes_in_newly_trained_model():
allele=allele,
peptides=df.peptide.values,
affinities=df.measurement_value.values,
+ verbose=0,
)
predict_and_check("HLA-A*01:01", "EVDPIGHLY", predictor=predictor)
@@ -157,6 +158,7 @@ def test_class1_affinity_predictor_a0205_memorize_training_data():
allele=allele,
peptides=df.peptide.values,
affinities=df.measurement_value.values,
+ verbose=0,
)
predictor.calibrate_percentile_ranks(num_peptides_per_length=1000)
ic50_pred = predictor.predict(df.peptide.values, allele=allele)
diff --git a/test/test_class1_neural_network.py b/test/test_class1_neural_network.py
index da83940b..36c02378 100644
--- a/test/test_class1_neural_network.py
+++ b/test/test_class1_neural_network.py
@@ -13,6 +13,7 @@ from mhcflurry.class1_neural_network import Class1NeuralNetwork
from mhcflurry.downloads import get_path
from mhcflurry.common import random_peptides
+
def test_class1_neural_network_a0205_training_accuracy():
# Memorize the dataset.
hyperparameters = dict(
@@ -80,7 +81,7 @@ def test_class1_neural_network_a0205_training_accuracy():
dense_layer_l1_regularization=0.0,
dropout_probability=0.0)
predictor2 = Class1NeuralNetwork(**hyperparameters2)
- predictor2.fit(df.peptide.values, df.measurement_value.values)
+ predictor2.fit(df.peptide.values, df.measurement_value.values, verbose=0)
eq_(predictor.network().to_json(), predictor2.network().to_json())
--
GitLab