# Imports
import torch
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
import numpy as np 
import seaborn as sns

DATASET_URL = "http://faculty.marshall.usc.edu/gareth-james/ISL/Advertising.csv"
DATA_FILENAME = "Advertising.csv"
TARGET_COLUMN = 'sales'

# Download the data
download_url(DATASET_URL, '.')
df = pd.read_csv(DATA_FILENAME)
df.head()