From 1bfa1882379a098e19247d5489883abd9cd7bacd Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Sat, 7 Sep 2019 16:15:24 -0400
Subject: [PATCH] fix
---
.travis.yml | 1 +
mhcflurry/class1_neural_network.py | 3 +++
2 files changed, 4 insertions(+)
diff --git a/.travis.yml b/.travis.yml
index d1ec644f..696dbd59 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -17,6 +17,7 @@ before_install:
- conda update -q conda
# Useful for debugging any issues with conda
- conda info -a
+ - free -m
addons:
apt:
packages:
diff --git a/mhcflurry/class1_neural_network.py b/mhcflurry/class1_neural_network.py
index 86a0cdf5..48db82f9 100644
--- a/mhcflurry/class1_neural_network.py
+++ b/mhcflurry/class1_neural_network.py
@@ -1368,7 +1368,10 @@ class Class1NeuralNetwork(object):
m is the length of the vectors used to represent amino acids
"""
from keras.models import clone_model
+ import tensorflow as tf
import keras.backend as K
+ K.get_session().run(tf.global_variables_initializer())
+
reshaped = allele_representations.reshape(
(allele_representations.shape[0], -1))
original_model = self.network()
--
GitLab