所以我基本上从适用于此模型的tensorflow教程中复制了代码:
它试图对神经网络建模以识别“楼梯”形状,如下所示:

(来源:gormanalysis.com)
import numpy as np
import tensorflow as tf
import _pickle as cPickle
with open("var_x.txt", "rb") as fp: # Unpickling
var_x = cPickle.load(fp)
with open("var_y.txt", "rb") as fp: # Unpickling
var_y = cPickle.load(fp)
# Declare list of features, we only have one real-valued feature
def model_fn(features, labels, mode):
# Build a linear model and predict values
W = tf.get_variable("W", [4], dtype=tf.float64)
b = tf.get_variable("b", [1], dtype=tf.float64)
y = tf.sigmoid( W*features['x'] + b)
# Loss sub-graph
loss …Run Code Online (Sandbox Code Playgroud)