# For Google Colab
!pip install git+https://github.com/Kaggle/kaggle-api.git --upgrade
import os
credentials = {"username":"","key":""}
os.environ['KAGGLE_USERNAME']=credentials["username"]
os.environ['KAGGLE_KEY']=credentials["key"]

# For Google Colab
#!kaggle datasets download -d paultimothymooney/chest-xray-pneumonia
#!unzip chest-xray-pneumonia.zip

import os
import torch
import torchvision
import tarfile
import torch.nn as nn
import numpy as np
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.metrics import roc_curve, auc
from PIL import Image, ImageOps
import torch.nn.functional as F
from torchvision.datasets.utils import download_url
import torchvision.models as models
from torchvision.datasets import ImageFolder
from torch.utils.data import Dataset, DataLoader, WeightedRandomSampler
import torchvision.transforms as T
from torch.utils.data import random_split
from torchvision.utils import make_grid
import matplotlib.pyplot as plt
from tqdm.notebook import tqdm
import seaborn as sns
%matplotlib inline

DATA_DIR = '../input/chest-xray-pneumonia/chest_xray'
#DATA_DIR = 'chest_xray/chest_xray'

TRAIN_DIR = DATA_DIR + '/train'                           # Contains training images
TEST_DIR = DATA_DIR + '/test'                             # Contains test images
VAL_DIR = DATA_DIR + '/val'                               # Contains test images