individual.py 文件源码

python
阅读 28 收藏 0 点赞 0 评论 0

项目:GAKeras 作者: PetraVidnerova 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号