def createNetwork(self):
model = Sequential()
firstlayer = True
for l in self.layers:
if firstlayer:
model.add(Dense(l.size, input_shape=self.input_shape))
firstlayer = False
else:
model.add(Dense(l.size))
model.add(Activation(l.activation))
if l.dropout > 0:
model.add(Dropout(l.dropout))
# final part
model.add(Dense(self.noutputs))
if Config.task_type == "classification":
model.add(Activation('softmax'))
model.compile(loss=Config.loss,
optimizer=RMSprop())
return model
评论列表
文章目录