Learn practical skills, build real-world projects, and advance your career
Created 3 years ago
This notebook is a cleaner version of the notebook https://jovian.ai/aakashns/03-logistic-regression with far less text and a succinct view.
The Model and Evaluation
import torch
import torchvision
import torchvision.transforms as transforms
from torchvision.datasets import MNIST#get dataset
dataset = MNIST(root='data/', train=True, transform=transforms.ToTensor(), download=True)
test_dataset = MNIST(root='data/', train=False, transform=transforms.ToTensor())#split dataset
from torch.utils.data import random_split
train_ds, val_ds = random_split(dataset, [50000, 10000])