当前位置：首页 > news >正文

tensorflow-线性回归python入门

news 2025/7/6 20:58:19

读入库

构造数据

建立训练和测试数据

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

开始测试

sin函数回归

读入库

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 datax_train,x_test,y_train,y_test = train_test_split(X,Y,test_size=0.1)# 建立 neural network from the first layer to last layermodel = 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()# 畫出 dataplt.scatter(X, Y)#X_test2=[-1,1]y_pred2 = model.predict(X)  # Y predictplt.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()