使用model.fit_generator()可视化/绘制keras指标

时间:2018-03-27 23:47:38

标签: python model keras visualization data-fitting

在keras中使用model.fit_generator()对我来说很有用,但是我想要准确,丢失等实时可视化,就像使用model.fit()一样。 无法找到有关如何使用model.fit_generator()函数执行此操作的任何说明。 我模型的训练方法如下:

def train_model(model, args, X_train, X_valid, y_train, y_valid):

    checkpoint = ModelCheckpoint('model-{epoch:03d}.h5',
                                 monitor='val_loss',
                                 verbose=0,
                                 save_best_only=args.save_best_only,
                                 mode='auto')

    model.compile(optimizer=Adam(lr=args.learning_rate), loss='mean_squared_error', metrics=['accuracy'])


    model.fit_generator(batch_generator(args.data_dir, X_train, y_train, args.batch_size, True),
                        args.samples_per_epoch,
                        args.nb_epoch,
                        max_q_size=1,
                        validation_data=batch_generator(args.data_dir, X_valid, y_valid, args.batch_size, False),
                        nb_val_samples=len(X_valid),
                        callbacks=[checkpoint],
                        verbose=1)

感谢您的帮助

0 个答案:

没有答案