PyTorch'ta кандайча классификатор түзүп, үйрөтүү керек?

PyTorch'ta klassifikator yaratmak we ony öwretmek üçin birnäçe ädimleri ýerine ýetirmeli:

1. Modeli kesgitlemek — nn.Module-den miras alýan synp döretmek we arxitekturany beýan etmek.
2. Maglumatlary taýýarlamak — maglumatlary ýükleýiň we DataLoader-a öwürüň.
3. Ýitgi funksiýasy we optimizatory kesgitlemek.
4. Modeli öwretmek — maglumatlar boýunça aýlan, çykaryşy hasapla, ýalňyşlygy sanaw, yzyna ýaýradma amala aşyr we agramlary täzeläň.

10 synplar üçin ýönekeý klassifikator mysaly:

import torch
import torch.nn as nn
import torch.optim as optim

class SimpleClassifier(nn.Module):
    def __init__(self):
        super().__init__()
        self.fc = nn.Linear(784, 10)  # mysal üçin 28x28 şekiller

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

model = SimpleClassifier()
criterion = nn.CrossEntropyLoss()
optimizer = optim.SGD(model.parameters(), lr=0.01)

# Öwretmek üçin mysal döngüsi
for epoch in range(num_epochs):
    for inputs, labels in dataloader:
        outputs = model(inputs)
        loss = criterion(outputs, labels)
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()

Bu kody goşmaça gatlaklar, aktivasiýa funksiýalary, maglumat işlemek we ş.m. bilen giňeldip bolýar.