From 624150da81e6529f0c9a353dcd5ed460f1b11fdf Mon Sep 17 00:00:00 2001 From: Francois Vieille Date: Thu, 14 Mar 2019 10:46:35 +0100 Subject: [PATCH] fonction loss --- iss/models/SimpleConvAutoEncoder.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/iss/models/SimpleConvAutoEncoder.py b/iss/models/SimpleConvAutoEncoder.py index c184475..9d72b93 100644 --- a/iss/models/SimpleConvAutoEncoder.py +++ b/iss/models/SimpleConvAutoEncoder.py @@ -75,6 +75,11 @@ class SimpleConvAutoEncoder(AbstractAutoEncoderModel): picture_dec = self.decoder_model(self.encoder_model(picture)) self.model = Model(picture, picture_dec) - optimizer = Adam(lr = 0.001, beta_1=0.9, beta_2=0.999, epsilon=None, decay=0.0, amsgrad=False) + optimizer = Adam(lr = self.lr, beta_1=0.9, beta_2=0.999, epsilon=None, decay=0.0, amsgrad=False) - self.model.compile(optimizer = optimizer, loss = 'binary_crossentropy') + def my_loss(picture, picture_dec): + loss = K.mean(K.binary_crossentropy(picture, picture_dec)) + return loss + + # self.model.compile(optimizer = optimizer, loss = 'binary_crossentropy') + self.model.compile(optimizer = optimizer, loss = my_loss)