Learn practical skills, build real-world projects, and advance your career
  1. Life expectancy prediction using linear regression (minimal)

Using the WHO Life Expectancy Dataset

# Uncomment and run the commands below if imports fail
# !conda install numpy pytorch torchvision cpuonly -c pytorch -y
# !pip install matplotlib --upgrade --quiet
!pip install jovian --upgrade --quiet
# Imports
import torch
import jovian
import torchvision
import torch.nn as nn
import pandas as pd
import matplotlib.pyplot as plt
import torch.nn.functional as F
from torchvision.datasets.utils import download_url
from torch.utils.data import DataLoader, TensorDataset, random_split
# Hyperparameters
batch_size=64
learning_rate=1e-1


# Other constants
DATASET_URL = "https://github.com/THC-QA/who-life-expectancy-dataset/raw/master/Life%20Expectancy%20Data.csv"
DATA_FILENAME = "Life%20Expectancy%20Data.csv"
TARGET_COLUMN = 'Life expectancy '
input_size=19
output_size=1

Dataset & Data loaders