From 88c036eeb66e38a41eea58ec6e4ca61a59a26082 Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 23 Jan 2020 16:41:51 -0500
Subject: [PATCH] fix
---
mhcflurry/class1_cleavage_neural_network.py | 10 +++-------
1 file changed, 3 insertions(+), 7 deletions(-)
diff --git a/mhcflurry/class1_cleavage_neural_network.py b/mhcflurry/class1_cleavage_neural_network.py
index aae3120f..53cc117d 100644
--- a/mhcflurry/class1_cleavage_neural_network.py
+++ b/mhcflurry/class1_cleavage_neural_network.py
@@ -170,13 +170,9 @@ class Class1CleavageNeuralNetwork(object):
Update self.network_json and self.network_weights properties based on
this instances's neural network.
"""
- network = self.network()
- if network is None:
- self.network_json = None
- self.network_weights = None
- else:
- self.network_json = network.to_json()
- self.network_weights = network.get_weights()
+ if self._network is not None:
+ self.network_json = self._network.to_json()
+ self.network_weights = self._network.get_weights()
def fit(
self,
--
GitLab