在Tensorflow中还原和预测

钢铁侠

我正在跟踪此链接以进行癌症预测。现在,在完成培训和测试阶段之后,我想提供新数据作为输入并希望进行预测。为此,我正在保存模型并还原它以获取预测,但是由于

ValueError: Cannot feed value of shape (31,) for Tensor 'Placeholder_1:0', which has shape '(?, 31)'

下面是我的代码:

saver = tf.train.Saver()
sampletest = [-0.24222039 -0.75688274 -0.26264569 -0.75637054 -0.7154845  -0.55675554 -0.51883267 -0.69442359 -0.87362527 -1.46135011 -0.05206671 -0.2790065 -0.28614862 -0.1934161  -0.38264881 -0.1295509   0.05817795 -0.32080093-0.64650773 -0.19383338 -0.14508449 -0.74260509 -0.66173979 -0.73123076-0.68635871 -0.78697688 -0.4790055  -0.71702336 -0.90543288 -1.1197415-0.41889736]
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())

    for epoch in range(training_epochs):
        for batch in range(int(n_samples / batch_size)):
            batch_x = input_X[batch * batch_size: (1 + batch) * batch_size]
            batch_y = input_Y[batch * batch_size: (1 + batch) * batch_size]
            print(batch_x[0])

            sess.run([optimizer], feed_dict={x: batch_x,
                                             y_: batch_y,
                                             pkeep: training_dropout})
            saver.save(sess,'.\cancer_model')

        # Display logs after every 10 epochs
        if (epoch) % display_step == 0:
            train_accuracy, newCost = sess.run([accuracy, cost],
                                               feed_dict={x: input_X, y_: input_Y, pkeep: training_dropout})

            valid_accuracy, valid_newCost = sess.run([accuracy, cost],
                                                     feed_dict={x: input_X_valid, y_: input_Y_valid, pkeep: 1})

            print("Epoch:", epoch, "Acc =", "{:.5f}".format(train_accuracy), "Cost =", "{:.5f}".format(newCost),
                  "Valid_Acc =", "{:.5f}".format(valid_accuracy), "Valid_Cost = ", "{:.5f}".format(valid_newCost))

            # Record the results of the model
            accuracy_history.append(train_accuracy)
            cost_history.append(newCost)
            valid_accuracy_history.append(valid_accuracy)
            valid_cost_history.append(valid_newCost)

            # If the model does not improve after 15 logs, stop the training.
            if valid_accuracy < max(valid_accuracy_history) and epoch > 100:
                stop_early += 1
                if stop_early == 15:
                    break
            else:
                stop_early = 0

    print("Optimization Finished!")

with tf.Session() as sess:
    saver = tf.train.import_meta_graph('.\Cancer_Model\cancer_model.meta')
    saver.restore(sess, tf.train.latest_checkpoint('.\Cancer_Model'))
    prediction = sess.run(y4,feed_dict={x:sampletest})
    print(prediction)

请帮我解决一下这个。

Jdehesa

问题在于您的模型需要大量示例,而您只是给出一个示例。尝试更换:

prediction = sess.run(y4, feed_dict={x: sampletest})

带有:

prediction = sess.run(y4, feed_dict={x: [sampletest]})

然后,您将获得一个prediction带有单个元素的“批量”结果

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章