您好,欢迎来到测品娱乐。
搜索
您的当前位置:首页tensorflow2.0 使用tf.data.Dataset创建模型多输入

tensorflow2.0 使用tf.data.Dataset创建模型多输入

来源:测品娱乐

以下为tensorflow源码:

train_dataset = tf.data.Dataset.from_tensor_slices(
    (
        {"img_input": img_data, "ts_input": ts_data},
        {"score_output": score_targets, "class_output": class_targets},
    )
)
train_dataset = train_dataset.shuffle(buffer_size=1024).batch()

model.fit(train_dataset, epochs=1)

在使用tf.data.Dataset.from_tensor_slices传参时要注意,有多个输入时,传进去的格式也要是字典,对应model.fit的传参。

大概整理下:

  tf.data.Dataset.from_tensor_slices model.fit
单个输入输出 x,y x,y
多个输入单个输出 {"input_x1":x1,"input_x2":x2},{

因篇幅问题不能全部显示,请点此查看更多更全内容

Copyright © 2019- cepb.cn 版权所有 湘ICP备2022005869号-7

违法及侵权请联系:TEL:199 18 7713 E-MAIL:2724546146@qq.com

本站由北京市万商天勤律师事务所王兴未律师提供法律服务