deblan-mirror/smart-iss-posts
1
0
Fork 0
mirror of https://github.com/prise6/smart-iss-posts synced 2024-04-24 18:20:28 +02:00
Code Issues Releases Wiki Activity

fonction loss

Browse source
This commit is contained in:
Francois Vieille 2019-03-14 10:46:35 +01:00
parent 9c76c82920
commit 624150da81
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
iss/models/SimpleConvAutoEncoder.py
View file

@ -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)