Tensorflow针对不同的训练输入参数返回相同的结果

时间:2017-07-17 18:42:14

标签: r tensorflow prediction

我想在R中实现具有张量流的预测函数,但无论输入参数是什么,预测结果总是相同的。

我试图非常接近官方tutorial

我的训练数据是在7个变量的数据框架中形成的。第一个是结果。结果是0或1。

我的完整代码是:

sess <- tf$InteractiveSession()

x <- tf$placeholder(tf$float32, shape(NULL,6L))
y_ <- tf$placeholder(tf$float32, shape(NULL,2L))

W <- tf$Variable(tf$zeros(shape(6L, 1L)))
b <- tf$Variable(tf$zeros(shape(2L)))

sess$run(tf$global_variables_initializer())
y <- tf$nn$softmax(tf$matmul(x,W) + b)
cross_entropy <- tf$reduce_mean(-tf$reduce_sum(y_ * tf$log(y), reduction_indices=1L))
optimizer <- tf$train$GradientDescentOptimizer(0.5)
train_step <- optimizer$minimize(cross_entropy)

i = 1
while (i < (nrow(training_data)-20)) {
    print(i)
    batch_ys <- matrix(c(training_data[i:(i+19),1], abs(training_data[i:(i+19),1]-1)), nrow=20)
    batch_xs <- matrix(c(training_data[i:(i+19),2],training_data[i:(i+19),3],training_data[i:(i+19),4],training_data[i:(i+19),5],training_data[i:(i+19),6],training_data[i:(i+19),7]), nrow=20, ncol=6)

    sess$run(train_step, feed_dict = dict(x = batch_xs, y_ = batch_ys))
    i = i + 20
}

使用训练数据的第一项进行简单检查

# Simple verification
for (j in 1:30){
    test_data <- c(training_data[j,2],training_data[j,3],training_data[j,4],training_data[j,5],training_data[j,6],training_data[j,7])
    test_data <- matrix(test_data, nrow = 1, ncol = 6)
    feed_dict = dict(x= test_data)
    print('############')
    print(sess$run(y,feed_dict)) # this is always the same
    print(training_data[j,1])   
}

我预计预测取决于输入,但它会返回:

[1] "############"
     [,1] [,2] [,3] [,4] [,5] [,6]
[1,]    2    2    4   74    5    2
      [,1]      [,2]
[1,] 0.0657808 0.9342192
[1] 1
[1] "############"
     [,1] [,2] [,3] [,4] [,5] [,6]
[1,]    0    1    5   61    2    3
      [,1]      [,2]
[1,] 0.0657808 0.9342192
[1] 0
[1] "############"
     [,1] [,2] [,3] [,4] [,5] [,6]
[1,]    3    2    6   85    5    4
      [,1]      [,2]
[1,] 0.0657808 0.9342192
[1] 0

我做错了什么?

谢谢,

的Bastian

1 个答案:

答案 0 :(得分:0)

似乎问题在于这一行:

W <- tf$Variable(tf$zeros(shape(6L, 1L)))

当您使用零初始化权重矩阵时,所有渐变也将为零。因此,即使经过一些训练迭代,W仍然是零的矩阵。解决方案是使用小的随机值来初始化该矩阵。