Learn practical skills, build real-world projects, and advance your career

Predicting pneumonia using logistic regression and Deep learning models

!pip install jovian --upgrade --quiet
project_name = 'final-course-project'
import torch
import jovian
import torchvision
import torch.nn as nn
import matplotlib.pyplot as plt
import torch.nn.functional as F
import torchvision.transforms as transforms
from torchvision.datasets import MNIST
from torch.utils.data import random_split
from torch.utils.data import DataLoader
# Hyperparmeters
batch_size = 128
learning_rate = 0.001
image_size = 64*64

# Other constants
input_size = 64*64*3
num_classes = 2