【问题标题】:How to extract train and validation sets in Keras?如何在 Keras 中提取训练集和验证集?
【发布时间】:2018-02-15 23:43:51
【问题描述】:

我在keras 中实现了一个神经网络,结构如下:

model = Sequential([... layers ...])
model.compile(optimizer=..., loss=...)
hist=model.fit(x=X,y=Y, validation_split=0.1, epochs=100)

有没有办法从modelhist 中提取训练集和验证集?也就是说,我想知道XY中的哪些索引用于训练,哪些用于验证。

【问题讨论】:

    标签: machine-learning keras


    【解决方案1】:

    Keras splits the dataset

    split_at = int(x[0].shape * (1-validation_split))
    

    进入训练和验证部分。因此,如果您有 n 样本,则第一个 int(n*(1-validation_split)) 样本将是训练样本,其余的是验证集。

    如果你想拥有更多的控制权,你可以自己拆分数据集,并通过参数validation_data传递验证数据集:

    model.fit(train_x, train_y, …, validation_data=(validation_x, validation_y))
    

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 2018-09-05
      • 2017-06-13
      • 2016-09-13
      • 2019-05-01
      • 2020-10-01
      • 1970-01-01
      • 1970-01-01
      • 2016-10-14
      相关资源
      最近更新 更多