【发布时间】:2016-08-24 03:20:13
【问题描述】:
我正在尝试创建一个程序,该程序将使用Tensorflow 将一个点分类为1 或0。我正在尝试围绕该图的中心创建一个椭圆形,其中蓝点是:
椭圆形中的所有内容都应归类为1,其他所有内容都应归类为0。在上图中,蓝点是1s,红点是0s。
但是,每次我尝试对一个点进行分类时,它总是选择1,即使它是我训练它使用的一个点,说它是0。
我的问题很简单:为什么猜测总是1,我做错了什么或者应该采取不同的方式来解决这个问题?这是我在没有教程的情况下尝试的第一个机器学习问题,所以我对这些东西真的不太了解。
如果您能提供任何帮助,我将不胜感激,谢谢!
这是我的代码:
#!/usr/bin/env python3
import tensorflow as tf
import numpy
import matplotlib.pyplot as plt
training_in = numpy.array([[0, 0], [1, 1], [2, 0], [-2, 0], [-1, -1], [-1, 1], [-1.5, 1], [3, 3], [3, 0], [-3, 0], [0, -3], [-1, 3], [1, -2], [-2, -1.5]])
training_out = numpy.array([1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0])
def transform_data(x):
return [x[0], x[1], x[0]**2, x[1]**2, x[0]*x[1]]
new_training_in = numpy.apply_along_axis(transform_data, 1, training_in)
feature_count = new_training_in.shape[1]
x = tf.placeholder(tf.float32, [None, feature_count])
y = tf.placeholder(tf.float32, [None, 1])
W = tf.Variable(tf.zeros([feature_count, 1]))
b = tf.Variable(tf.zeros([1]))
guess = tf.nn.softmax(tf.matmul(x, W) + b)
cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(tf.matmul(x, W) + b, y))
opti = tf.train.GradientDescentOptimizer(0.01).minimize(cost)
init = tf.initialize_all_variables()
sess = tf.Session()
sess.run(init)
for i in range(1000):
for (item_x, item_y) in zip(new_training_in, training_out):
sess.run(opti, feed_dict={ x: [item_x], y: [[item_y]]})
print(sess.run(W))
print(sess.run(b))
plt.plot(training_in[:6, 0], training_in[:6, 1], 'bo')
plt.plot(training_in[6:, 0], training_in[6:, 1], 'rx')
results = sess.run(guess, feed_dict={ x: new_training_in })
for i in range(training_in.shape[0]):
xx = [training_in[i:,0]]
yy = [training_in[i:,1]]
res = results[i]
# this always prints `[ 1.]`
print(res)
# uncomment these lines to see the guesses
# if res[0] == 0:
# plt.plot(xx, yy, 'c+')
# else:
# plt.plot(xx, yy, 'g+')
plt.show()
【问题讨论】:
标签: python machine-learning tensorflow logistic-regression