nn.py 文件源码

def load_model(data):
    '''
    Load keras model.
    '''
    model = Sequential()
    model.add(Dense(data.shape[1], activation='relu',
        input_dim=data.shape[1], kernel_constraint=maxnorm(3)))
    model.add(Dropout(0.25))
    #model.add(Dense(16, activation='relu', kernel_constraint=maxnorm(3)))
    #model.add(Dropout(0.25))
    model.add(Dense(1, activation='sigmoid'))

    model.compile(optimizer='RMSprop', loss='binary_crossentropy',
        metrics=['accuracy'])

    return model