如何在tensorflow中使用自定义数据集?

时间:2016-05-16 05:52:34

标签: machine-learning tensorflow deep-learning tensorboard

我最近开始学习张量流。我试图输入我的自定义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()

如何在张量流中使用此图作为输入向量?

1 个答案:

答案 0 :(得分:2)

您必须使用tf.placeholder函数(see the doc):

# 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,))

然后,您可以在张量流代码中使用x_node和y_node(例如,使用x_node作为神经网络的输入并尝试预测y_node)。
然后,在使用sess.run()时,您必须使用x参数提供输入数据yfeed_dict

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