如何用Keras记录训练样本的id模特。适合吗?

2024-04-26 01:18:01 发布

您现在位置:Python中文网/ 问答频道 /正文

我目前有再现性问题,尽管I set the seeds。我知道模型是以同样的方式初始化的(通过用h5dump和meld检查model.save("initial.h5"))。你知道吗

接下来我要检查训练样本是否按相同顺序使用。因此,我想记录他们。你知道吗

我通过

model.fit(dataset['train']['X'],
          dataset['train']['y'],
          epochs=cfg['model']['nb_epochs'],
          batch_size=cfg['model']['batch_size'],
          validation_split=cfg['model']['validation_split'],
          callbacks=[checkpoint], class_weight=cw)

我还可以添加dataset['train']['id']。我想得到一个txt文件,其中包含正在使用的ID列表,例如,对于32个批量大小、765个训练数据集长度和5个历元,我希望在txt文件中有765*5=3825行,其中每个ID大约出现5次,前32个元素是第一批的ID。你知道吗

有可能吗?你知道吗


Tags: 文件thetxtidsizemodelbatchtrain