2

我有两个特征 X1 和 X2 并想预测 Y。

训练数据

X1  X2   Y
11  610 676
52  557 120
78  491 964
77  380 722
24  464 837
86  532 601
99  580 452
10  539 200
88  507 756

如何使用 tf.layers.dense 和 tf.data.Dataset 做到这一点?请不要使用 feed_dict,因为它不再推荐。

我搜索了同时使用 tf.layers.dense 和 tf.data.Dataset 但没有任何可靠的代码示例。

训练模型后,我想预测以下测试数据的 Y。预测测试数据是否需要 dataset.make_initializable_iterator()?

测试数据

X1  X2
36  608
48  170
58  455
99  927
13  431
61  312
34  763
94  626
14  193

如果模型架构包括批量标准化、正则化和 dropout,但现在不是优先事项,它将是一个插件。

谢谢

4

1 回答 1

1

这篇来自向数据科学的教程很好地解释了数据集 API 的用法
https://towardsdatascience.com/how-to-use-dataset-in-tensorflow-c758ef9e4428

相关的 Jupyter 笔记本
https://github.com/FrancescoSaverioZuppichini/Tensorflow-Dataset-Tutorial/blob/master/dataset_tutorial.ipynb

于 2018-03-19T16:31:24.657 回答