【发布时间】:2017-04-10 09:06:05
【问题描述】:
我正在使用sklearn 的波士顿住房数据集(一个 506x13 矩阵)进行多元线性回归。我计划使用所有数据对其进行训练,然后“插入”一个随机数据,例如boston_dataset.data[39],然后查看损失。但是当我打印结果时,我得到的只是NaN。这是我的代码。
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import load_boston
np.set_printoptions(suppress=True)
boston = load_boston()
m = boston.data.shape[0] - 1
bt_unfixed = np.transpose(boston.data)
bt = np.insert(bt_unfixed, 0, 1)
Y = tf.placeholder(tf.float64, name='Y___')
X = tf.placeholder(tf.float64, [1, 13], name='X_____')
#print X.shape
W = tf.Variable(tf.zeros([13, 1]), name='weights')
b = tf.Variable(0.5, name='bias')
hypothesis = tf.add(tf.matmul(X, tf.cast(W, tf.float64)), tf.cast(b, tf.float64))
loss = tf.reduce_sum(tf.square(hypothesis - Y)) / (2 * m)
optimizer = tf.train.GradientDescentOptimizer(0.01)
train_op = optimizer.minimize(loss)
with tf.Session() as sess:
sess.run(tf.initialize_all_variables())
for i in range(0, 500):
for (x, y) in zip(boston.data, boston.target):
sess.run(train_op, feed_dict={X:x.reshape(1, 13), Y:y})
if (i + 1)%50 == 0:
print "Ran " + str(i) + "times\nW=" +str(sess.run(W)) + "\nb=" +str(sess.run(b))
print "Done!\n"
print "Running test...\n"
t = sess.run(cost, feed_dict={X:boston.data[504], Y:boston.target.data[504]})
print "loss =" + str(t) + "Real value" + str(boston.target.data[504]) + "Pred " +str(sess.run(hypothesis, feed_dict={X:boston.data[504]}))
谢谢!也请随时添加任何建议
【问题讨论】:
标签: machine-learning scikit-learn tensorflow linear-regression