model_combined.py 文件源码

def build_model(self, models, input_dims, output_dim):
        keras_models = [None] * len(models)
        for i, m in enumerate(models):
            m.build_model(input_dims[i], output_dim)
            keras_models[i] = m.model
        merged = Merge(keras_models, mode='concat')
        self.model.add(merged)
        self.model.add(Dense(output_dim, activation='softmax'))
        self.model.compile(optimizer='adam',
                           loss='categorical_crossentropy',
                           metrics=['accuracy'])