model.py 文件源码

def get_model():
    model = Sequential()

    model.add(Dense(1024, init='normal', input_dim = 460))
    model.add(LeakyReLU(0.3))
    model.add(Dropout(0.5))

    model.add(Dense(1024, init='normal'))
    model.add(LeakyReLU(0.3))
    model.add(Dropout(0.5))

    model.add(Dense(512, init='normal'))
    model.add(LeakyReLU(0.3))
    model.add(Dropout(0.5))

    model.add(Dense(1, init='normal'))
    return model