用tensorflow拟合二次函数

最新推荐文章于 2024-08-06 00:30:14 发布

一只荣

最新推荐文章于 2024-08-06 00:30:14 发布

阅读量680

点赞数

分类专栏：机器学习算法实现

本文链接：https://blog.csdn.net/qq_35455503/article/details/102637610

版权

机器学习算法实现专栏收录该内容

2 篇文章 0 订阅

订阅专栏

#!/usr/bin/env python
# encoding: utf-8
'''
@author: shibaorong
@license: (C) Copyright 2019, Node Supply Chain Manager Corporation Limited.
@contact: diamond_br@163.com
@software: pycharm
@file: quadratic_regression.py
@time: 2019/10/14 19:17
@desc:
'''
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt


def add_layer(input,insize,outsize):
    W=tf.Variable(tf.random_normal([insize,outsize]))
    b=tf.Variable(tf.zeros([1,outsize]))
    wx_b=tf.matmul(input,W)+b
    wx_bact=tf.nn.relu(wx_b)

    return wx_bact

if __name__=='__main__':

    x1=np.array([-5,-4,-3,-2,-1,0,1,2,3,4,5],dtype=np.float).reshape([-1,1])
    y1=np.array([2,7,9,12,13,14,14,13,10,8,4],dtype=np.float).reshape([-1,1])

    x=tf.placeholder(dtype=tf.float32,shape=[None,1])
    y=tf.placeholder(dtype=tf.float32,shape=[None,1])

    layer1=add_layer(x,1,11)
    layer2=add_layer(layer1,11,1)

    loss=tf.reduce_sum(tf.square(layer2-y))

    globals_step=tf.Variable(0,trainable=False)
    init_LR=0.0001
    learning_rate=tf.train.exponential_decay(init_LR,globals_step,100000,0.5)
    optimizer=tf.train.GradientDescentOptimizer(learning_rate)
    train_op=optimizer.minimize(loss)

    init=tf.global_variables_initializer()
    with tf.Session() as sess:
        sess.run(init)

        while(1):
            sess.run(train_op,feed_dict={x:x1,y:y1})
            print(sess.run(loss,feed_dict={x:x1,y:y1}))



        layer2_val=sess.run(layer2,feed_dict={x:x1,y:y1})
        plt.figure()
        plt.scatter(x1,y1)
        plt.plot(x1,layer2_val,'r-',lw=5)
        plt.show()