tensorflow-线性回归python入门

最新推荐文章于 2024-10-13 10:05:58 发布

三十度角阳光的问候

最新推荐文章于 2024-10-13 10:05:58 发布

阅读量919

点赞数 7

文章标签： python tensorflow 线性回归

本文链接：https://blog.csdn.net/weixin_43802726/article/details/142004480

版权

读入库

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
import time

构造数据

X = np.array([0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0])
Y = np.array([0.1, 0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8, 0.9, 1.0])

建立训练和测试数据

import sklearn.model_selection as sk
X_train, X_test, Y_train, Y_test =sk.train_test_split(X,Y,test_size=0.2, random_state = 42)

创建第一层到最后一层的神经网络

model = tf.contrib.keras.models.Sequential()
model.add(tf.contrib.keras.layers.Dense(units=1, activation=tf.nn.relu,input_dim=1))
model.summary()
# 開始搭建 model
# mse = mean square error
# sgd = stochastic gradient descent
model.compile(loss='mse',optimizer='sgd',metrics=['accuracy'])

model.fit(X_test, Y_test,
          epochs=4000,
          batch_size=len(Y_test))

开始测试

print("start testing")

cost = model.evaluate(X_test, Y_test)

print("test cost: {}".format(cost))

weights, biases = model.layers[0].get_weights()

print("Weights = {}, bias = {}".format(weights,biases))

打印测试结果

# 印出測試的結果
Y_pred = model.predict(X_test)
# 畫出 data
plt.scatter(X,Y, label='X,Y')
plt.scatter(X_test, Y_test, label='X_test, Y_test')
plt.scatter(X_test, Y_pred, label='pred')
# 畫出 線
x2 = np.linspace(0,1,100)
print(biases[0])
print(weights[0])

y2 =(weights[0]*x2+biases[0])
plt.plot(x2, y2, '-r', label='weights')

plt.show()

sin函数回归

np.random.seed(int(time.time())) 生成随机数
num=100 随机数100个
X = np.linspace(-4,4,num) 进行线性等分
np.random.shuffle(X) 打乱
Y = 0.1*np.sin(X) 计算变量

#!/usr/bin/env python
# -*- coding=utf-8 -*-
import tensorflow as tf
from sklearn.model_selection import train_test_split
import numpy as np
import matplotlib.pyplot as plt
import time

# 製造 data (共200筆)
np.random.seed(int(time.time()))
num=100
X = np.linspace(-4,4,num)
np.random.shuffle(X)
Y = 0.1*np.sin(X)

# 建立 trainig 與 testing data

x_train,x_test,y_train,y_test = train_test_split(X,Y,test_size=0.1)

# 建立 neural network from the first layer to last layer

model = tf.keras.models.Sequential([
   tf.keras.layers.Dense(units=100, activation=tf.nn.tanh, input_dim=1),
   tf.keras.layers.Dense(units=100, activation=tf.nn.tanh),
   tf.keras.layers.Dense(units=1, activation=tf.nn.tanh),
])

# 除了第一層以外，定義第二層以上時，不需要定義 input dimension，因為第二層 input 就是第一層的 input

# 開始搭建 model
# mse = mean square error
# sgd = stochastic gradient descent
model.compile(loss='mse',optimizer='sgd', metrics=['acc'])


# training
print("start training")
for step in range(20000):
    cost = model.train_on_batch(x_train, y_train)  #
    if step % 20 == 0:
        #print("train cost{}".format(cost))
        W, b = model.layers[0].get_weights()
        print("step{} Weights = {}, bias = {} train cost{}".format(step,W, b, cost))
        plt.cla()
        # 畫出 data
        plt.scatter(X, Y)
        #X_test2=[-1,1]
        y_pred2 = model.predict(X)  # Y predict
        plt.scatter(X, y_pred2, color='blue')
        plt.text(0, -0.05, 'epoch: %d  ,cost=%.2f '
                 % (step,cost[0]), fontdict={'size': 10, 'color': 'red'})
        plt.pause(0.01)

其他数据训练示例

#!/usr/bin/env python

# -*- coding=utf-8 -*-

import tensorflow as tf

from sklearn.model_selection import train_test_split

import numpy as np

import matplotlib.pyplot as plt

import time

import pandas as pd

from tensorflow.keras.datasets import boston_housing

import matplotlib.pyplot as plt

(x_train, y_train), (x_test, y_test) = boston_housing.load_data()