From 809c271a743ae75064440a0c65465af8d3334070 Mon Sep 17 00:00:00 2001 From: prise6 Date: Sun, 10 Mar 2019 18:49:10 +0100 Subject: [PATCH] config activation --- iss/models/SimpleAutoEncoder.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/iss/models/SimpleAutoEncoder.py b/iss/models/SimpleAutoEncoder.py index 0b011d0..06139be 100644 --- a/iss/models/SimpleAutoEncoder.py +++ b/iss/models/SimpleAutoEncoder.py @@ -15,6 +15,7 @@ class SimpleAutoEncoder(AbstractModel): super().__init__(save_directory, model_name) + self.activation = config['activation'] self.input_shape = (config['input_height'], config['input_width'], config['input_channel']) self.lr = config['learning_rate'] self.build_model() @@ -39,7 +40,7 @@ class SimpleAutoEncoder(AbstractModel): # x = Conv2D(1, (3, 3), activation = 'relu', padding = 'same', name = 'dec_conv_1')(encoded) # x = UpSampling2D((2, 2))(x) # x = Flatten()(x) - x = Dense(np.prod(input_shape), activation = 'softmax')(layer_5) + x = Dense(np.prod(input_shape), activation = self.activation)(layer_5) decoded = Reshape((input_shape))(x) self.model = Model(picture, decoded)