anywords.py 文件源码

def train_model(self, batch_size, epochs, path):

        filepath = "./traindata/checkpoints/weights-{epoch:02d}.hdf5"
        checkpoint = ModelCheckpoint(filepath,
                                     monitor='loss',
                                     verbose=1,
                                     save_best_only=True,
                                     mode='min')
        callbacks_list = [checkpoint]

        self.model.fit(self.x_data,
                       self.y_data,
                       batch_size=batch_size,
                       epochs=epochs,
                       callbacks=callbacks_list)
        self.model.save_weights(path)