main.py 文件源码

def model_EEDS(input_col, input_row):
    _input = Input(shape=(input_col, input_row, 1), name='input')
    EES = model_EES(input_col, input_row)(_input)
    EED = model_EED(input_col, input_row)(_input)
    _EEDS = merge(inputs=[EED, EES], mode='sum')

    model = Model(input=_input, output=_EEDS)
    Adam = adam(lr=0.001)
    model.compile(optimizer=Adam, loss='mean_squared_error', metrics=['mean_squared_error'])
    return model