Learn practical skills, build real-world projects, and advance your career
Updated 4 years ago
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