1. 程式人生 > 其它 >【專案實戰】kaggle產品分類挑戰

【專案實戰】kaggle產品分類挑戰

多分類特徵的學習

這裡還是b站劉二大人的視訊課程式碼,視訊連結:https://www.bilibili.com/video/BV1Y7411d7Ys?p=9
相關注釋已經標明瞭(就當是筆記),因此在這裡不多贅述,今天的主要目的還是Kaggle的題目

import torch
from torchvision import transforms  # 影象處理工具
from torchvision import datasets
from torch.utils.data import DataLoader
import torch.nn.functional as F
import torch.optim as optim

batch_size = 64
transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.1307, ), (0.3081, ))
])
# 可以把傳入的影象變成一個數值在0到1之間的張量,這裡的均值和標準差都是算好的
train_dataset = datasets.MNIST(root='../dataset/mnist/',
                               train=True,
                               download=True,
                               transform=transform)
train_loader = DataLoader(train_dataset,
                          shuffle=True,
                          batch_size=batch_size)
text_dataset = datasets.MNIST(root='../dataset/mnist/',
                              train=False,
                              download=True,
                              transform=transform)
test_loader = DataLoader(text_dataset,
                         shuffle=False,
                         batch_size=batch_size)


class Net(torch.nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.l1 = torch.nn.Linear(784, 512)
        self.l2 = torch.nn.Linear(512, 256)
        self.l3 = torch.nn.Linear(256, 128)
        self.l4 = torch.nn.Linear(128, 64)
        self.l5 = torch.nn.Linear(64, 10)

    def forward(self, x):
        x = x.view(-1, 784)
        x = F.relu(self.l1(x))
        x = F.relu(self.l2(x))
        x = F.relu(self.l3(x))
        x = F.relu(self.l4(x))
        return self.l5(x)


model = Net()
criterion = torch.nn.CrossEntropyLoss()
# 使用交叉熵損失他的作用是儘量保持當前梯度的變化方向。
# 沒有動量的網路可以視為一個質量很輕的棉花團,風往哪裡吹就往哪裡走,一點風吹草動都影響他,四處跳
# 動不容易學習到更好的區域性最優。沒有動力來源的時候可能又不動了。加了動量就像是棉花變成了鐵球,
# 咕嚕咕嚕的滾在引數空間裡,很容易闖過鞍點,直到最低點。可以參照指數滑動平均。優化效果是梯度二階
# 導數不會過大,優化更穩定,也可以看做效果接近二階方法,但是計算容易的多。其實本質應該是對引數加了約束
optimizer = optim.SGD(model.parameters(), lr=0.01, momentum=0.5)  # 這裡是一個衝量


def train(epoch):
    running_loss = 0.0
    for batch_idx, data in enumerate(train_loader, 0):
        inputs, target = data
        optimizer.zero_grad()
        # 前饋反饋和更新
        outputs = model(inputs)
        loss = criterion(outputs, target)
        loss.backward()
        optimizer.step()

        running_loss += loss.item()  # 累積的loss拿入
        if batch_idx % 300 == 299:    # 每300輪輸出一次
            print('[%d, %5d] loss: %.3f' % (epoch+1, batch_idx+1, running_loss/300))
            running_loss = 0.0


def test():
    correct = 0  # 正確的數量
    total = 0   # 總數
    with torch.no_grad():
        for data in test_loader:   # 拿出每一行裡面最大值的下標
            images, labels = data
            outputs = model(images)
            _, predicted = torch.max(outputs.data, dim=1)
            total += labels.size(0)
            correct += (predicted == labels).sum().item()  # 計算猜對的數量
            print('Accuracy on text set: %d %%' % (100*correct/total))


if __name__ == '__main__':
    for epoch in range(10):
        train(epoch)
        test()
# 到97就跑不上去了,畢竟這就是一個很簡單的前饋網路

然後這裡是結果

賽題及閱讀

本題的kaggle網址在此:https://www.kaggle.com/competitions/otto-group-product-classification-challenge
然後資料集一個train一個text,train,這個挑戰主要是給一堆商品資料,這些商品分為9類,每個商品有93個特徵,需要你來進行一個模型學習,在測試集中完成分類
原文翻譯如下:
在本次比賽中,我們為20多萬種產品提供了一個包含93個功能的資料集。我們的目標是建立一個預測模型,能夠區分我們的主要產品類別。獲獎模型將是開源的。

資料處理

首先我們的資料長成這樣


顯然我們的target裡面的就是我們要學習的標籤值,但是他是一個string型別,這個我們可以進行轉換

def class2num(brands):
    brands_out = []
    brand_in = ['class1', 'class2', 'class3', 'class4', 'class5', 'class6', 'class7', 'class8', 'class9']
    for brand in brands:   //迴圈的讀傳入的資料
        brands_out.append(brand_in.index(brand))  ##獲取當前值在brand_in的索引,巧妙的讓他數字化
    return brands_out

然後再定義處理資料的函式

class ProductData(Dataset):
    def __init__(self, filepath):
        xy = pd.read_csv(filepath)
        brand = xy['target']
        self.len = xy.shape[0]
        self.x_data = torch.tensor(np.array(xy)[:, 1:-1].astype(float))  # 把xy轉換成np的array陣列然後讀取裡面的資料,並進行資料轉換
        self.y_data = class2num(brand)

    def __getitem__(self, index):
        return self.x_data[index], self.y_data[index]

    def __len__(self):
        return self.len

再傳入資料,定義小批量梯度下降的規模

train_data = ProductData('train.csv')
train_loader = DataLoader(dataset=train_data, batch_size=64, shuffle=True, num_workers=0)

模型設計

再開始設計模型,因為我們的最後分類是9,而特徵有93個,所以需要把93降到9
這裡是我們定義的模型

class Net(torch.nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.l1 = torch.nn.Linear(93, 72)
        self.l2 = torch.nn.Linear(72, 36)
        self.l3 = torch.nn.Linear(36, 18)
        self.l4 = torch.nn.Linear(18, 9)

    def forward(self, x):
        x = x.view(-1, 93)
        x = F.relu(self.l1(x))
        x = F.relu(self.l2(x))
        x = F.relu(self.l3(x))
        return self.l4(x)

同時在Net的下面我還決定按照昨天的程式碼編寫思路,寫上測試的函式

 def test(self, x):
        with torch.no_grad():
            x = F.relu(self.l1(x))
            x = F.relu(self.l2(x))
            x = F.relu(self.l3(x))
            x = F.relu(self.l4(x))
            _, predict = torch.max(x, dim=1)
# 程式碼中一個獨立的下劃線,表示這個變數不重要一個獨立的下劃線,它也是一個變數名,只不過它比較特殊,當你使用下劃線作為變數名時,就代表你告訴大家,這個變數不重要,僅僅佔個位置,可以忽略,後面不會再使用它。
            y = pd.get_dummies(predict)   # 把predict轉換成獨熱向量
            return y

例項化模型,並確定損失函式和優化器

model = Net()
criterion = torch.nn.CrossEntropyLoss()
optimizer = torch.optim.SGD(model.parameters(), lr=0.01, momentum=0.5)

訓練

if __name__ == '__main__':
    for epoch in range(200):
        running_loss = 0.0   
        for batch_idx, data in enumerate(train_loader, 0):  //用batch_idx, data獲取enumerate的值
            inputs, target = data
            inputs = inputs.float()
            optimizer.zero_grad()
            outputs = model(inputs)      //  放進模型
            loss = criterion(outputs, target)
            loss.backward()
            optimizer.step()

            running_loss += loss.item()  # 累積的loss拿入
            if batch_idx % 300 == 299:    # 每300輪輸出一次
                print('[%d, %5d] loss: %.3f' % (epoch+1, batch_idx+1, running_loss/300))
                running_loss = 0.0

這就是小批量梯度下降的訓練寫法
訓練結果如下

測試

按照要求來測試資料集並且輸出一個csv

test_data = pd.read_csv('test.csv')   # 讀取測試集
test_inputs = torch.tensor(np.array(test_data)[:, 1:].astype(float))   # 讀取特徵
out = model.test(test_inputs.float())          # 傳入模型並且用out來接受得到的值
lables=['Class_1', 'Class_2', 'Class_3', 'Class_4', 'Class_5', 'Class_6', 'Class_7', 'Class_8', 'Class_9']
out.columns = lables   # 用lables作為標籤
out.insert(0, 'id', test_data['id'])   # 對應的標籤插入值,insert() 函式用於將指定物件插入列表的指定位置
output = pd.DataFrame(out)    # 輸出成一個dataframe
output.to_csv('my_predict.csv', index=False)  # 輸出為csv檔案

放進kaggle看看

分數不太行嗷
改變一下動量看看(改成0.1)

稍微好一點
以下是完整程式碼

import numpy as np
import pandas as pd
from torch.utils.data import Dataset, DataLoader
import torch
import torch.nn.functional as F
import torch.optim as optim


def class2num(brands):
        brands_out = []
        brand_in = ['Class_1', 'Class_2', 'Class_3', 'Class_4', 'Class_5', 'Class_6', 'Class_7', 'Class_8', 'Class_9']
        for brand in brands:
            brands_out.append(brand_in.index(brand))
        return brands_out


class ProductData(Dataset):
    def __init__(self, filepath):
        xy = pd.read_csv(filepath)
        brand = xy['target']
        self.len = xy.shape[0]
        self.x_data = torch.tensor(np.array(xy)[:, 1:-1].astype(float))  # 把xy轉換成np的array陣列然後讀取裡面的資料,並進行資料轉換
        self.y_data = class2num(brand)

    def __getitem__(self, index):
        return self.x_data[index], self.y_data[index]

    def __len__(self):
        return self.len


train_data = ProductData('train.csv')
train_loader = DataLoader(dataset=train_data, batch_size=64, shuffle=True, num_workers=0)


class Net(torch.nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.l1 = torch.nn.Linear(93, 72)
        self.l2 = torch.nn.Linear(72, 36)
        self.l3 = torch.nn.Linear(36, 18)
        self.l4 = torch.nn.Linear(18, 9)

    def forward(self, x):
        x = x.view(-1, 93)
        x = F.relu(self.l1(x))
        x = F.relu(self.l2(x))
        x = F.relu(self.l3(x))
        return self.l4(x)

    def test(self, x):
        with torch.no_grad():
            x = F.relu(self.l1(x))
            x = F.relu(self.l2(x))
            x = F.relu(self.l3(x))
            x = F.relu(self.l4(x))
            _, predict = torch.max(x, dim=1)
# 程式碼中一個獨立的下劃線,表示這個變數不重要一個獨立的下劃線,它也是一個變數名,只不過它比較特殊,當你使用下劃線作為變數名時,就代表你告訴大家,這個變數不重要,僅僅佔個位置,可以忽略,後面不會再使用它。
            y = pd.get_dummies(predict)   # 把predict轉換成獨熱向量
            return y


model = Net()
criterion = torch.nn.CrossEntropyLoss()
optimizer = optim.SGD(model.parameters(), lr=0.01, momentum=0.1)

if __name__ == '__main__':
    for epoch in range(200):
        running_loss = 0.0
        for batch_idx, data in enumerate(train_loader, 0):
            inputs, target = data
            inputs = inputs.float()
            optimizer.zero_grad()
            outputs = model(inputs)
            loss = criterion(outputs, target)
            loss.backward()
            optimizer.step()

            running_loss += loss.item()  # 累積的loss拿入
            if batch_idx % 300 == 299:    # 每300輪輸出一次
                print('[%d, %5d] loss: %.3f' % (epoch+1, batch_idx+1, running_loss/300))
                running_loss = 0.0

# 測試
test_data = pd.read_csv('test.csv')
test_inputs = torch.tensor(np.array(test_data)[:, 1:].astype(float))
out = model.test(test_inputs.float())
lables=['Class_1', 'Class_2', 'Class_3', 'Class_4', 'Class_5', 'Class_6', 'Class_7', 'Class_8', 'Class_9']
out.columns = lables
out.insert(0, 'id', test_data['id'])
output = pd.DataFrame(out)
output.to_csv('my_predict.csv', index=False)

全梯度

故技重施,搞一下直接梯度下降看看,訓練5w遍

import numpy as np
import pandas as pd
import torch
import torch.nn.functional as F
import torch.optim as optim


def class2num(brands):
        brands_out = []
        brand_in = ['Class_1', 'Class_2', 'Class_3', 'Class_4', 'Class_5', 'Class_6', 'Class_7', 'Class_8', 'Class_9']
        for brand in brands:
            brands_out.append(brand_in.index(brand))
        return brands_out


xy = pd.read_csv('train.csv')
brand = xy['target']
len = xy.shape[0]
x_data = torch.tensor(np.array(xy)[:, 1:-1].astype(float))  # 把xy轉換成np的array陣列然後讀取裡面的資料,並進行資料轉換
y_data = torch.tensor(class2num(brand))


class Net(torch.nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.l1 = torch.nn.Linear(93, 72)
        self.l2 = torch.nn.Linear(72, 36)
        self.l3 = torch.nn.Linear(36, 18)
        self.l4 = torch.nn.Linear(18, 9)

    def forward(self, x):
        x = x.view(-1, 93)
        x = F.relu(self.l1(x))
        x = F.relu(self.l2(x))
        x = F.relu(self.l3(x))
        return self.l4(x)

    def test(self, x):
        with torch.no_grad():
            x = F.relu(self.l1(x))
            x = F.relu(self.l2(x))
            x = F.relu(self.l3(x))
            x = F.relu(self.l4(x))
            _, predict = torch.max(x, dim=1)
# 程式碼中一個獨立的下劃線,表示這個變數不重要一個獨立的下劃線,它也是一個變數名,只不過它比較特殊,當你使用下劃線作為變數名時,就代表你告訴大家,這個變數不重要,僅僅佔個位置,可以忽略,後面不會再使用它。
            y = pd.get_dummies(predict)   # 把predict轉換成獨熱向量
            return y


model = Net()
criterion = torch.nn.CrossEntropyLoss()
optimizer = optim.SGD(model.parameters(), lr=0.01, momentum=0.1)

if __name__ == '__main__':
    for epoch in range(50000):
        x_data = x_data.float()
        optimizer.zero_grad()
        outputs = model(x_data)
        loss = criterion(outputs, y_data)
        print(epoch, loss.item())
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()


# 測試
test_data = pd.read_csv('test.csv')
test_inputs = torch.tensor(np.array(test_data)[:, 1:].astype(float))
out = model.test(test_inputs.float())
lables=['Class_1', 'Class_2', 'Class_3', 'Class_4', 'Class_5', 'Class_6', 'Class_7', 'Class_8', 'Class_9']
out.columns = lables
out.insert(0, 'id', test_data['id'])
output = pd.DataFrame(out)
output.to_csv('my_predict.csv', index=False)

這次訓練速度肉眼可見的慢,慢的一批

下降的也非常緩慢,基本可以判斷是碰到區域性最優了,畢竟剛剛那個200次就loss就0.3了
最後我們再用隨機梯度下降試試,只需要把batchsize改成1即可
慢的就離譜。。。。

半天就跑這麼點
那沒事了
結束今天的學習!