deblan-mirror/smart-iss-posts
1
0
Fork 0
mirror of https://github.com/prise6/smart-iss-posts synced 2024-04-27 03:21:51 +02:00
Code Issues Releases Wiki Activity

config activation

Browse source
This commit is contained in:
prise6 2019-03-10 18:49:10 +01:00
parent a6969908fd
commit 809c271a74
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
iss/models/SimpleAutoEncoder.py
View file

@ -15,6 +15,7 @@ class SimpleAutoEncoder(AbstractModel):
super().__init__(save_directory, model_name) super().__init__(save_directory, model_name)
self.activation = config['activation']
self.input_shape = (config['input_height'], config['input_width'], config['input_channel']) self.input_shape = (config['input_height'], config['input_width'], config['input_channel'])
self.lr = config['learning_rate'] self.lr = config['learning_rate']
self.build_model() 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 = Conv2D(1, (3, 3), activation = 'relu', padding = 'same', name = 'dec_conv_1')(encoded)
# x = UpSampling2D((2, 2))(x) # x = UpSampling2D((2, 2))(x)
# x = Flatten()(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) decoded = Reshape((input_shape))(x)
self.model = Model(picture, decoded) self.model = Model(picture, decoded)