TrainData_PT_recur.py 文件源码

def model(input_shapes, nclasses):
        from keras.layers import Input
        from keras.layers.core import Masking
        x_global  = Input(shape=input_shapes[0])
        x = Dense(10, activation='relu',kernel_initializer='lecun_uniform')(x_global)
        for _ in range(6):
            x = Dense(10, activation='relu',kernel_initializer='lecun_uniform')(x)

        predictions = Dense(
            nclasses, activation='softmax',
            kernel_initializer='lecun_uniform',
            name='classification_out'
        )(x)
        return Model(inputs=x_global, outputs=predictions)