From 89cc638efc09bb6913cd25306f5a6f3915923df0 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Mon, 15 Jul 2019 13:08:39 -0400
Subject: [PATCH] fix
---
test/expensive_test_pretrain_optimizable.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/test/expensive_test_pretrain_optimizable.py b/test/expensive_test_pretrain_optimizable.py
index 59cd8105..3a261c65 100644
--- a/test/expensive_test_pretrain_optimizable.py
+++ b/test/expensive_test_pretrain_optimizable.py
@@ -38,7 +38,7 @@ HYPERPARAMTERS = {
'early_stopping': True, 'init': 'glorot_uniform',
'layer_sizes': [1024, 512], 'learning_rate': None,
'locally_connected_layers': [], 'loss': 'custom:mse_with_inequalities',
- 'max_epochs': 1, 'min_delta': 0.0, 'minibatch_size': 128,
+ 'max_epochs': 0, 'min_delta': 0.0, 'minibatch_size': 128,
'optimizer': 'rmsprop', 'output_activation': 'sigmoid', 'patience': 20,
'peptide_allele_merge_activation': '',
'peptide_allele_merge_method': 'concatenate',
@@ -82,8 +82,9 @@ def test_optimizable():
},
)
(network,) = predictor.neural_networks
- pretrain_val_loss = network.fit_info[-1]['training_info']["val_loss"][-1]
- print(predictor)
+ print(predictor, network)
+ print(network.fit_info)
+ pretrain_val_loss = network.fit_info[0]["val_loss"][-1]
print(pretrain_val_loss)
numpy.testing.assert_array_less(pretrain_val_loss, 0.1)
--
GitLab