Python torch構建神經網路(基礎)

2021-09-25 14:04:20 字數 996 閱讀 4698

neu=torch.nn.sequential(  #建立神經網路,儲存到變數中

torch.nn.linear(input_size,hidden_size),

torch.nn.sigmoid(),

torch.nn.linear(hidden_size,output_size),

)

cost=torch.nn.mseloss()
optimizer=torch.optim.sgd(

neu.parameters(), #神經網路的各項引數,包括權重和偏置

lr=0.01 #學習率

)

#神經網路訓練迴圈

for i in range(2000):

batch_loss= #記錄每一批的損失

#start和end分別是提取乙個batch上標和下標

for start in range(0, len[x], batch_size):

end=start+batch_size if start+batch_sizexx=variable(torch.floattensor(x[start:end])) #一批特徵屬性資料

yy=variable(torch.floattensor(y[start:end])) #一批目標屬性資料

predict=neu(xx) #模型**

loss=cost(predict,yy)

optimizer.zero_grad() #將優化器儲存的哪些引數的梯度設定為0

loss.backward() #開始反向傳播計算所有梯度值

optimizer.step() #優化器開始執行一步更新所有的引數

#每隔100步輸出一下損失值(loss)

if i%100==0:

print(i,np.mean(batch_loss))

Keras 神經網路構建學習

keras 是簡單易用 高效強大的神經網路庫,底層計算可基於 tensorflow 或 theano 平台實現。本篇將詳細介紹 keras 模型構建的具體步驟。通過閱讀本篇內容您將了解到 keras 模型構建的主要步驟 keras 神經網路搭建的一般過程 keras 模型構建主要包括5個步驟 定義 ...

用pytorch構建神經網路

在神經網路模型中之前,要對資料進行一系列的預處理,如果是型別變數,可使用one hot編碼 對於數值型別,可進行標準化,讓其屬性值在0左右波動 import torch import numpy as np import pandas as pd from torch.autograd import...

使用pytorch構建神經網路

介紹 從學習神經網路到現在時間也不短了,由於個人數學能力有限,用numpy構建神經網路,實屬力不從心,但還是將神經網路的基本步驟理清了,然後開始學習用pytorch搭建神經網路。以下記錄構建神經網路的簡單方法。import numpy as np import torch n,d in,h,d ou...