微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

TFRecord 有多个 X,如何通过 X 来拟合模型

如何解决TFRecord 有多个 X,如何通过 X 来拟合模型

我有一个 TFRecord 有多个 X 变化,结构如下:

image,image_pos,image_grayscale,image_pos_grayscale,label = next(iter(test_dataset))

但是我只想传递一个X来拟合模型,而不是全部,怎么做?

history = mdl.fit(  train_dataset,# this will pass all X including variations
                        validation_data=valid_dataset,# this will pass all X including variations
                        #batch_size=batch_size,#shuffle=True,#False,callbacks=[early_stop],epochs=epochs,verbose=2)

我想要类似的东西:

history = mdl.fit(  x=image_train,y=label_train,# only pass image
                        validation_data=(image_valid,label_valid),#batch_size=batch_size,verbose=2)

我想要的另一个模型:

history = mdl.fit(  x=(image_train,image_pos_train),# pass image and image_pos
                        validation_data=((image_valid,verbose=2)

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。