machine-learning - 如何在张量流中使用自定义数据集？

Question

我最近开始学习 tensorflow。我正在尝试输入我的自定义 python 代码作为训练数据。我已经生成了随机指数信号，并希望网络从中学习。这是我用来生成信号的代码-

import matplotlib.pyplot as plt
import random
import numpy as np

lorange= 1
hirange= 10
amplitude= random.uniform(-10,10)
t= 10
random.seed()
tau=random.uniform(lorange,hirange)
x=np.arange(t)

plt.xlabel('t=time")
plt.ylabel('x(t)')
plt.plot(x, amplitude*np.exp(-x/tau))
plt.show()

如何将此图用作张量流中的输入向量？

Answer 1

您必须使用tf.placeholder功能（请参阅文档）：

# Your input data
x = np.arange(t)
y = amplitude*np.exp(-x/tau)

# Create a corresponding tensorflow node
x_node = tf.placeholder(tf.float32, shape=(t,))
y_node = tf.placeholder(tf.float32, shape=(t,))

然后，您可以在 tensorflow 代码中使用 x_node 和 y_node （例如x_node用作神经网络的输入并尝试进行预测y_node）。
然后在使用时，sess.run()您必须提供输入数据x并y带有feed_dict参数：

with tf.Session() as sess: 
    sess.run([...], feed_dict={x_node: x, y_node: y})