tflearn学习2 线性回归

pip install tensorflow tflearn

""" 线性回归实例 """

from __future__ import absolute_import, division, print_function

import tflearn

# 回归数据
X = [3.3,4.4,5.5,6.71,6.93,4.168,9.779,6.182,7.59,2.167,7.042,10.791,5.313,7.997,5.654,9.27,3.1]
Y = [1.7,2.76,2.09,3.19,1.694,1.573,3.366,2.596,2.53,1.221,2.827,3.465,1.65,2.904,2.42,2.94,1.3]

# 线性回归模型
input_ = tflearn.input_data(shape=[None])
linear = tflearn.single_unit(input_)
regression = tflearn.regression(linear, optimizer='sgd', loss='mean_square',
                                metric='R2', learning_rate=0.01)
m = tflearn.DNN(regression)
m.fit(X, Y, n_epoch=1000, show_metric=True, snapshot_epoch=False)

print("\n回归结果:")
print("Y = " + str(m.get_weights(linear.W)) +
      "*X + " + str(m.get_weights(linear.b)))

print("\n测试预测:\n x = 3.2, 3.3, 3.4:")
print("\n y = " + str(m.predict([3.2, 3.3, 3.4])))
# 应输出（不准确） y = [1.5315033197402954, 1.5585315227508545, 1.5855598449707031]