From 8a19a894aaa776dff58122d57d748f8c3233803b Mon Sep 17 00:00:00 2001
From: Alex Rubinsteyn <alex.rubinsteyn@gmail.com>
Date: Mon, 9 May 2016 19:04:21 -0400
Subject: [PATCH] only train for 200 epochs in unit test
---
test/test_imputation.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/test/test_imputation.py b/test/test_imputation.py
index 9dfea81f..e73894aa 100644
--- a/test/test_imputation.py
+++ b/test/test_imputation.py
@@ -96,7 +96,7 @@ def test_performance_improves_for_A0205_with_pretraining():
X_pretrain=X_index_imputed,
ic50_pretrain=ic50_imputed,
sample_weights_pretrain=sample_weights_imputed,
- n_training_epochs=500)
+ n_training_epochs=200)
ic50_pred_with_imputation = \
predictor_with_imputation.predict_ic50_for_kmer_encoded_array(X_index)
--
GitLab