Keras model.fit()参数详解

1、这是keras官方文档上关于model.fit()参数介绍。

fit( x=None, #输入的x值
     y=None, #输入的y标签值
     batch_size=None, #整数 ,每次梯度更新的样本数即批量大小。未指定,默认为32。
     epochs=1, #迭代次数
     verbose=1, #整数,代表以什么形式来展示日志状态
     callbacks=None, #回调函数,这个list中的回调函数将会在训练过程中的适当时机被调用,参考回调函数
     validation_split=0.0, #浮点数0-1之间,用作验证集的训练数据的比例。模型将分出一部分不会被训练的验证数据,并将在每一轮结束时评估这些验证数据的误差和任何其他模型指标。
     validation_data=None, #这个参数会覆盖 validation_split,即两个函数只能存在一个,它的输入为元组 (x_val,y_val),这作为验证数据。
     shuffle=True, #布尔值。是否在每轮迭代之前混洗数据
     class_weight=None,
     sample_weight=None, 
     initial_epoch=0, 
     steps_per_epoch=None, #一个epoch包含的步数(每一步是一个batch的数据送入),当使用如TensorFlow数据Tensor之类的输入张量进行训练时,默认的None代表自动分割,即数据集样本数/batch样本数。
     validation_steps=None, #在验证集上的step总数,仅当steps_per_epoch被指定时有用。
     validation_freq=1, #指使用验证集实施验证的频率。当等于1时代表每个epoch结束都验证一次
     max_queue_size=10,
     workers=1,
     use_multiprocessing=False
   )

2、在实际应用中的model.fit()调用。

history = model.fit(train_x,
                    train_y,
                    batch_size=batch_size,
                    epochs=100,
                    #validation_split=0.125,
                    validation_data=[val_x,val_y],
                    shuffle= True,
                    verbose=1,
                    callbacks=[checkpoint],
                    validation_freq=1)
  • x

  • y

  • batch_size:整数 ,每次梯度更新的样本数。未指定,默认为32。

  • epochs:整数,训练模型迭代次数。

  • verbose:日志展示,整数
                   0 :为不在标准输出流输出日志信息

  •                1 :显示进度条


  •                2 :每个epoch输出一行记录

  • Keras model.fit()参数详解_第1张图片

你可能感兴趣的:(keras学习,python)