From 39b1b9c8646d88b0c43649c25b0c7a3db5f420fc Mon Sep 17 00:00:00 2001
From: Tim O'Donnell <timodonnell@gmail.com>
Date: Thu, 12 Dec 2019 14:55:45 -0500
Subject: [PATCH] loss_reduction = "sum"
---
mhcflurry/class1_presentation_neural_network.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/mhcflurry/class1_presentation_neural_network.py b/mhcflurry/class1_presentation_neural_network.py
index 970c5286..e580e715 100644
--- a/mhcflurry/class1_presentation_neural_network.py
+++ b/mhcflurry/class1_presentation_neural_network.py
@@ -453,7 +453,7 @@ class Class1PresentationNeuralNetwork(object):
allele_representations_hash = self.set_allele_representations(
allele_representations)
- loss_reduction = "sum_over_batch_size"
+ loss_reduction = "sum"
self.network.compile(
loss=[
affinities_loss.get_keras_loss(reduction=loss_reduction),
@@ -846,4 +846,4 @@ class Class1PresentationNeuralNetwork(object):
instance.network = keras.models.model_from_json(network_json)
if weights is not None:
instance.network.set_weights(weights)
- return instance
\ No newline at end of file
+ return instance
--
GitLab