pytorch解決鳶尾花分類

半年前用numpy寫了個鳶尾花分類200行。。每一步計算都是手寫的  python構建bp神經網絡_鳶尾花分類

現在用pytorch簡單寫一遍，pytorch語法解釋請看上一篇pytorch搭建簡單網絡

import pandas as pd
import torch.nn as nn
import torch


class MyNet(nn.Module):
    def __init__(self):
        super(MyNet, self).__init__()
        self.fc = nn.Sequential(
            nn.Linear(4, 3),
            nn.Sigmoid(),
            nn.Linear(3, 3),
            nn.Sigmoid(),
            nn.Linear(3, 1),
        )
        self.mls = nn.MSELoss()
        self.opt = torch.optim.Adam(params=self.parameters(), lr=0.001)

    def get_data(self):
        inputs = []
        labels = []
        with open('flower.csv') as file:
            df = pd.read_csv(file, header=None)
            x = df.iloc[:, 0:4].values
            y = df.iloc[:, 4].values
            for i in range(len(x)):
                inputs.append(x[i])
            for j in range(len(y)):
                a = []
                a.append(y[j])
                labels.append(a)

        return inputs, labels

    def forward(self, inputs):
        out = self.fc(inputs)
        return out

    def train(self, x, label):
        out = self.forward(x)
        loss = self.mls(out, label)
        self.opt.zero_grad()
        loss.backward()
        self.opt.step()

    def test(self, x):
        return self.fc(x)


if __name__ == '__main__':
    net = MyNet()
    inputs, labels = net.get_data()
    for i in range(1000):
        for index, input in enumerate(inputs):
            # 這里不加.float()會報錯，可能是數據格式的問題吧
            input = torch.from_numpy(input).float()
            label = torch.Tensor(labels[index])
            net.train(input, label)
    # 簡單測試一下
    c = torch.Tensor([[5.6, 2.7, 4.2, 1.3]])
    print(net.test(c))

運行結果趨近於0.5  正確，單純練一下pytorch，就沒有分訓練集，測試集

tensor([[0.5392]], grad_fn=<AddmmBackward>)

不用手寫反向傳播和梯度下降 是多么幸福一件事~