Learn practical skills, build real-world projects, and advance your career
Updated 3 years ago
Logistic Regression
# Imports
import torch
import torchvision
from torchvision.datasets import MNIST
# Download training dataset
dataset = MNIST(root='data/', download=True)
len(dataset)
60000
test_dataset = MNIST(root='data/', train=False)
len(test_dataset)
10000