如何在急切执行模式中替换tf.placeholder

我正在参考一个tf1的代码来开发一个模型。在参考代码中,他们使用了tf.placeholder来为模型提供输入,代码如下:

class model(object):    def __init__(self, lstm_size, batch_len, output_nodes, keep_prob, learning_rate=0.001):        self.inputs_ = tf.compat.v1.placeholder(tf.float32, shape=[batch_len, None, 512], name='lstm_inputs')        self.targets_ = tf.compat.v1.placeholder(tf.int32, [batch_len], name='lstm_targets')

这个模型的执行模式是图执行(tf.Graph)。

由于我想将代码转换并在急切执行模式下运行,但是tf.placeholder在急切执行模式下无法工作。我该如何替换tf.placeholder,同时不影响输入名称、数据类型和输入值的形状,以便适应急切执行模式?

再次强调,我需要将其转换为急切执行模式。


回答:

你可以使用tf.keras.Input()来替代tf.compat.v1.placeholder(),以便在急切执行模式下运行代码。

另外,请查看这个迁移指南,了解如何将TensorFlow 1.x代码转换为TensorFlow 2.x

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注