{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"pygments_lexer":"ipython3","nbconvert_exporter":"python","version":"3.6.4","file_extension":".py","codemirror_mode":{"name":"ipython","version":3},"name":"python","mimetype":"text/x-python"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"## Image Classification with Pytorch","metadata":{}},{"cell_type":"markdown","source":"This notebook is my first classification work and it could be look coarse but i think it will be helpful to pytorch real starter.","metadata":{}},{"cell_type":"markdown","source":"### there are 8 processes while doing this job.","metadata":{}},{"cell_type":"markdown","source":"1. import Library\n2. hyper parameter setting\n3. set seed and random value \n4. data load\n5. make our model\n6. set our loss function and optimizer\n7. train\n8. check our model performance or using test datasets in trained model","metadata":{}},{"cell_type":"markdown","source":"\n### 1. Import Library","metadata":{}},{"cell_type":"code","source":"#about torch...\nimport torch\nimport torch.nn as nn\nimport torch\nimport torch.optim as optim\nimport torch.nn.functional as F\n\nfrom torchvision import datasets, models, transforms\nfrom torch.utils.data import DataLoader, Dataset\n\n#using numpy\nimport numpy as np\n\n#for data load or save\nimport pandas as pd\n\n#visualize some datasets\nimport matplotlib.pyplot as plt\n\n#check our work directory\nimport os\n\n#to unzip datasets\nimport zipfile","metadata":{"_uuid":"8f2839f25d086af736a60e9eeb907d3b93b6e0e5","_cell_guid":"b1076dfc-b9ad-4769-8c92-a6c4dae69d19","execution":{"iopub.status.busy":"2021-08-14T09:54:25.579338Z","iopub.execute_input":"2021-08-14T09:54:25.579678Z","iopub.status.idle":"2021-08-14T09:54:27.223509Z","shell.execute_reply.started":"2021-08-14T09:54:25.579649Z","shell.execute_reply":"2021-08-14T09:54:27.222461Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### 2. Hyper parameters Setting","metadata":{}},{"cell_type":"code","source":"lr = 0.001 # learning_rate\nbatch_size = 100 # we will use mini-batch method\nepochs = 10 # How much to train a model","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:27.225123Z","iopub.execute_input":"2021-08-14T09:54:27.225431Z","iopub.status.idle":"2021-08-14T09:54:27.231082Z","shell.execute_reply.started":"2021-08-14T09:54:27.225396Z","shell.execute_reply":"2021-08-14T09:54:27.230076Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"3. Set seed and Random Value","metadata":{}},{"cell_type":"code","source":"device = 'cuda' if torch.cuda.is_available() else 'cpu'\n\ntorch.manual_seed(1234)\nif device =='cuda':\n    torch.cuda.manual_seed_all(1234)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:27.232552Z","iopub.execute_input":"2021-08-14T09:54:27.233134Z","iopub.status.idle":"2021-08-14T09:54:27.244795Z","shell.execute_reply.started":"2021-08-14T09:54:27.233089Z","shell.execute_reply":"2021-08-14T09:54:27.243786Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### 4. Load train,  test data","metadata":{}},{"cell_type":"markdown","source":"In this competition, we can't use datasets with only pandas\nwe should to unzip datasets. and we must check our working directory before unzipping","metadata":{}},{"cell_type":"markdown","source":"check a directory","metadata":{}},{"cell_type":"code","source":"os.listdir('../input/dogs-vs-cats-redux-kernels-edition')","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:27.246498Z","iopub.execute_input":"2021-08-14T09:54:27.246877Z","iopub.status.idle":"2021-08-14T09:54:27.258335Z","shell.execute_reply.started":"2021-08-14T09:54:27.246842Z","shell.execute_reply":"2021-08-14T09:54:27.257321Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"#### make a directory for our datasets after unzipping","metadata":{}},{"cell_type":"code","source":"os.makedirs('../data', exist_ok=True)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:27.262235Z","iopub.execute_input":"2021-08-14T09:54:27.262683Z","iopub.status.idle":"2021-08-14T09:54:27.266849Z","shell.execute_reply.started":"2021-08-14T09:54:27.262635Z","shell.execute_reply":"2021-08-14T09:54:27.265808Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"base_dir = '../input/dogs-vs-cats-redux-kernels-edition'\ntrain_dir = '../data/train'\ntest_dir = '../data/test'","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:27.269557Z","iopub.execute_input":"2021-08-14T09:54:27.269921Z","iopub.status.idle":"2021-08-14T09:54:27.277785Z","shell.execute_reply.started":"2021-08-14T09:54:27.269891Z","shell.execute_reply":"2021-08-14T09:54:27.27692Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"with zipfile.ZipFile(os.path.join(base_dir, 'train.zip')) as train_zip:\n    train_zip.extractall('../data')\n    \nwith zipfile.ZipFile(os.path.join(base_dir, 'test.zip')) as test_zip:\n    test_zip.extractall('../data')","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:27.279344Z","iopub.execute_input":"2021-08-14T09:54:27.279724Z","iopub.status.idle":"2021-08-14T09:54:48.677503Z","shell.execute_reply.started":"2021-08-14T09:54:27.279685Z","shell.execute_reply":"2021-08-14T09:54:48.676738Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"os.listdir(train_dir)[:5]","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:48.678627Z","iopub.execute_input":"2021-08-14T09:54:48.679102Z","iopub.status.idle":"2021-08-14T09:54:48.703001Z","shell.execute_reply.started":"2021-08-14T09:54:48.679058Z","shell.execute_reply":"2021-08-14T09:54:48.702019Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"import glob\n\ntrain_list = glob.glob(os.path.join(train_dir,'*.jpg'))\ntest_list = glob.glob(os.path.join(test_dir, '*.jpg'))","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:48.70414Z","iopub.execute_input":"2021-08-14T09:54:48.704567Z","iopub.status.idle":"2021-08-14T09:54:48.853647Z","shell.execute_reply.started":"2021-08-14T09:54:48.704526Z","shell.execute_reply":"2021-08-14T09:54:48.852841Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"len(train_list)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:48.854899Z","iopub.execute_input":"2021-08-14T09:54:48.855358Z","iopub.status.idle":"2021-08-14T09:54:48.861095Z","shell.execute_reply.started":"2021-08-14T09:54:48.855312Z","shell.execute_reply":"2021-08-14T09:54:48.860133Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"#### check our datasets","metadata":{}},{"cell_type":"code","source":"from PIL import Image\nrandom_idx = np.random.randint(1,25000,size=10)\n\nfig = plt.figure()\ni=1\nfor idx in random_idx:\n    ax = fig.add_subplot(2,5,i)\n    img = Image.open(train_list[idx])\n    plt.imshow(img)\n    i+=1\n\nplt.axis('off')\nplt.show()","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:48.862346Z","iopub.execute_input":"2021-08-14T09:54:48.862661Z","iopub.status.idle":"2021-08-14T09:54:49.987281Z","shell.execute_reply.started":"2021-08-14T09:54:48.862632Z","shell.execute_reply":"2021-08-14T09:54:49.986263Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_list[0].split('/')[-1].split('.')[0]","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:49.988594Z","iopub.execute_input":"2021-08-14T09:54:49.988882Z","iopub.status.idle":"2021-08-14T09:54:49.994921Z","shell.execute_reply.started":"2021-08-14T09:54:49.988853Z","shell.execute_reply":"2021-08-14T09:54:49.994171Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"int(test_list[0].split('/')[-1].split('.')[0])","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:49.99607Z","iopub.execute_input":"2021-08-14T09:54:49.996408Z","iopub.status.idle":"2021-08-14T09:54:50.008971Z","shell.execute_reply.started":"2021-08-14T09:54:49.996369Z","shell.execute_reply":"2021-08-14T09:54:50.008044Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"print(len(train_list), len(test_list))","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:50.010542Z","iopub.execute_input":"2021-08-14T09:54:50.011284Z","iopub.status.idle":"2021-08-14T09:54:50.018419Z","shell.execute_reply.started":"2021-08-14T09:54:50.011238Z","shell.execute_reply":"2021-08-14T09:54:50.017088Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"from sklearn.model_selection import train_test_split\ntrain_list, val_list = train_test_split(train_list, test_size=0.2)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:50.020268Z","iopub.execute_input":"2021-08-14T09:54:50.020782Z","iopub.status.idle":"2021-08-14T09:54:51.060883Z","shell.execute_reply.started":"2021-08-14T09:54:50.02074Z","shell.execute_reply":"2021-08-14T09:54:51.060041Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### Image Augumentation\n- we have to do this job before making our model. cuz it will prevent overfitting Possibility","metadata":{}},{"cell_type":"code","source":"#data Augumentation\ntrain_transforms =  transforms.Compose([\n        transforms.Resize((224, 224)),\n        transforms.RandomResizedCrop(224),\n        transforms.RandomHorizontalFlip(),\n        transforms.ToTensor(),\n    ])\n\nval_transforms = transforms.Compose([\n        transforms.Resize((224, 224)),\n        transforms.RandomResizedCrop(224),\n        transforms.RandomHorizontalFlip(),\n        transforms.ToTensor(),\n    ])\n\n\ntest_transforms = transforms.Compose([   \n    transforms.Resize((224, 224)),\n     transforms.RandomResizedCrop(224),\n    transforms.RandomHorizontalFlip(),\n    transforms.ToTensor()\n    ])\n","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.062833Z","iopub.execute_input":"2021-08-14T09:54:51.063287Z","iopub.status.idle":"2021-08-14T09:54:51.072427Z","shell.execute_reply.started":"2021-08-14T09:54:51.063244Z","shell.execute_reply":"2021-08-14T09:54:51.071248Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### Load datasets\n- this code is for load our image sets. you can find this code easily in pytorch.org , main page","metadata":{}},{"cell_type":"code","source":"class dataset(torch.utils.data.Dataset):\n    #가져와서 처리\n    def __init__(self,file_list,transform=None):\n        self.file_list = file_list\n        self.transform = transform\n        \n        \n    #dataset length\n    def __len__(self):\n        self.filelength = len(self.file_list)\n        return self.filelength\n    \n    #load an one of images\n    def __getitem__(self,idx):\n        img_path = self.file_list[idx]\n        img = Image.open(img_path)\n        img_transformed = self.transform(img)\n        \n        label = img_path.split('/')[-1].split('.')[0]\n        if label == 'dog':\n            label=1\n        elif label == 'cat':\n            label=0\n            \n        return img_transformed,label\n        ","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.074358Z","iopub.execute_input":"2021-08-14T09:54:51.074655Z","iopub.status.idle":"2021-08-14T09:54:51.087249Z","shell.execute_reply.started":"2021-08-14T09:54:51.074629Z","shell.execute_reply":"2021-08-14T09:54:51.086075Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"#### use torchvision.datasets","metadata":{}},{"cell_type":"code","source":"train_data = dataset(train_list, transform=train_transforms)\ntest_data = dataset(test_list, transform=test_transforms)\nval_data = dataset(val_list, transform=test_transforms)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.089641Z","iopub.execute_input":"2021-08-14T09:54:51.090652Z","iopub.status.idle":"2021-08-14T09:54:51.097142Z","shell.execute_reply.started":"2021-08-14T09:54:51.089922Z","shell.execute_reply":"2021-08-14T09:54:51.09631Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"train_loader = torch.utils.data.DataLoader(dataset = train_data, batch_size=batch_size, shuffle=True )\ntest_loader = torch.utils.data.DataLoader(dataset = test_data, batch_size=batch_size, shuffle=True)\nval_loader = torch.utils.data.DataLoader(dataset = val_data, batch_size=batch_size, shuffle=True)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.098143Z","iopub.execute_input":"2021-08-14T09:54:51.09847Z","iopub.status.idle":"2021-08-14T09:54:51.107259Z","shell.execute_reply.started":"2021-08-14T09:54:51.098439Z","shell.execute_reply":"2021-08-14T09:54:51.106164Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"print(len(train_data), len(train_loader))","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.108696Z","iopub.execute_input":"2021-08-14T09:54:51.109127Z","iopub.status.idle":"2021-08-14T09:54:51.119803Z","shell.execute_reply.started":"2021-08-14T09:54:51.109083Z","shell.execute_reply":"2021-08-14T09:54:51.118863Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"print(len(val_data), len(val_loader))","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.121021Z","iopub.execute_input":"2021-08-14T09:54:51.121474Z","iopub.status.idle":"2021-08-14T09:54:51.130509Z","shell.execute_reply.started":"2021-08-14T09:54:51.121429Z","shell.execute_reply":"2021-08-14T09:54:51.1295Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"#check our images shape\ntrain_data[0][0].shape","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.131526Z","iopub.execute_input":"2021-08-14T09:54:51.131812Z","iopub.status.idle":"2021-08-14T09:54:51.175877Z","shell.execute_reply.started":"2021-08-14T09:54:51.131785Z","shell.execute_reply":"2021-08-14T09:54:51.174681Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### build Model\n- 3 Convolution layer and 2 fully connected layer\n- batchNormalization for limit overfitting","metadata":{}},{"cell_type":"code","source":"class Cnn(nn.Module):\n    def __init__(self):\n        super(Cnn,self).__init__()\n        \n        self.layer1 = nn.Sequential(\n            nn.Conv2d(3,16,kernel_size=3, padding=0,stride=2),\n            nn.BatchNorm2d(16),\n            nn.ReLU(),\n            nn.MaxPool2d(2)\n        )\n        \n        self.layer2 = nn.Sequential(\n            nn.Conv2d(16,32, kernel_size=3, padding=0, stride=2),\n            nn.BatchNorm2d(32),\n            nn.ReLU(),\n            nn.MaxPool2d(2)\n            )\n        \n        self.layer3 = nn.Sequential(\n            nn.Conv2d(32,64, kernel_size=3, padding=0, stride=2),\n            nn.BatchNorm2d(64),\n            nn.ReLU(),\n            nn.MaxPool2d(2)\n        )\n        \n        \n        self.fc1 = nn.Linear(3*3*64,10)\n        self.dropout = nn.Dropout(0.5)\n        self.fc2 = nn.Linear(10,2)\n        self.relu = nn.ReLU()\n        \n        \n    def forward(self,x):\n        out = self.layer1(x)\n        out = self.layer2(out)\n        out = self.layer3(out)\n        out = out.view(out.size(0),-1)\n        out = self.relu(self.fc1(out))\n        out = self.fc2(out)\n        return out","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.177134Z","iopub.execute_input":"2021-08-14T09:54:51.177554Z","iopub.status.idle":"2021-08-14T09:54:51.195006Z","shell.execute_reply.started":"2021-08-14T09:54:51.177484Z","shell.execute_reply":"2021-08-14T09:54:51.19329Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"model = Cnn().to(device)\nmodel.train()","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.196933Z","iopub.execute_input":"2021-08-14T09:54:51.198126Z","iopub.status.idle":"2021-08-14T09:54:51.223088Z","shell.execute_reply.started":"2021-08-14T09:54:51.198078Z","shell.execute_reply":"2021-08-14T09:54:51.222069Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### set Loss function and optimizer","metadata":{}},{"cell_type":"code","source":"optimizer = optim.Adam(params = model.parameters(),lr=0.001)\ncriterion = nn.CrossEntropyLoss()","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.224451Z","iopub.execute_input":"2021-08-14T09:54:51.225076Z","iopub.status.idle":"2021-08-14T09:54:51.231247Z","shell.execute_reply.started":"2021-08-14T09:54:51.22503Z","shell.execute_reply":"2021-08-14T09:54:51.23005Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### Train our Network","metadata":{}},{"cell_type":"code","source":"epochs = 10\n\nfor epoch in range(epochs):\n    epoch_loss = 0\n    epoch_accuracy = 0\n    \n    for data, label in train_loader:\n        data = data.to(device)\n        label = label.to(device)\n        \n        output = model(data)\n        loss = criterion(output, label)\n        \n        optimizer.zero_grad()\n        loss.backward()\n        optimizer.step()\n        \n        acc = ((output.argmax(dim=1) == label).float().mean())\n        epoch_accuracy += acc/len(train_loader)\n        epoch_loss += loss/len(train_loader)\n        \n    print('Epoch : {}, train accuracy : {}, train loss : {}'.format(epoch+1, epoch_accuracy,epoch_loss))\n    \n    \n    with torch.no_grad():\n        epoch_val_accuracy=0\n        epoch_val_loss =0\n        for data, label in val_loader:\n            data = data.to(device)\n            label = label.to(device)\n            \n            val_output = model(data)\n            val_loss = criterion(val_output,label)\n            \n            \n            acc = ((val_output.argmax(dim=1) == label).float().mean())\n            epoch_val_accuracy += acc/ len(val_loader)\n            epoch_val_loss += val_loss/ len(val_loader)\n            \n        print('Epoch : {}, val_accuracy : {}, val_loss : {}'.format(epoch+1, epoch_val_accuracy,epoch_val_loss))","metadata":{"execution":{"iopub.status.busy":"2021-08-14T09:54:51.232698Z","iopub.execute_input":"2021-08-14T09:54:51.233327Z","iopub.status.idle":"2021-08-14T10:41:55.555065Z","shell.execute_reply.started":"2021-08-14T09:54:51.233283Z","shell.execute_reply":"2021-08-14T10:41:55.552733Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"dog_probs = []\nmodel.eval()\nwith torch.no_grad():\n    for data, fileid in test_loader:\n        data = data.to(device)\n        preds = model(data)\n        preds_list = F.softmax(preds, dim=1)[:, 1].tolist()\n        dog_probs += list(zip(list(fileid), preds_list))","metadata":{"execution":{"iopub.status.busy":"2021-08-14T10:41:55.560071Z","iopub.execute_input":"2021-08-14T10:41:55.560879Z","iopub.status.idle":"2021-08-14T10:43:36.92643Z","shell.execute_reply.started":"2021-08-14T10:41:55.560794Z","shell.execute_reply":"2021-08-14T10:43:36.925491Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"- In this Problem, we used softmax except sigmoid. \n- why? isn't it binary classification?\n- before i read overview, i didn't recognize that. read overview about this competition First!!","metadata":{}},{"cell_type":"code","source":"dog_probs.sort(key = lambda x : int(x[0]))\ndog_probs","metadata":{"execution":{"iopub.status.busy":"2021-08-14T10:43:36.927732Z","iopub.execute_input":"2021-08-14T10:43:36.928148Z","iopub.status.idle":"2021-08-14T10:43:36.993963Z","shell.execute_reply.started":"2021-08-14T10:43:36.928118Z","shell.execute_reply":"2021-08-14T10:43:36.993184Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"idx = list(map(lambda x: x[0],dog_probs))\nprob = list(map(lambda x: x[1],dog_probs))","metadata":{"execution":{"iopub.status.busy":"2021-08-14T10:43:36.995222Z","iopub.execute_input":"2021-08-14T10:43:36.995659Z","iopub.status.idle":"2021-08-14T10:43:37.00851Z","shell.execute_reply.started":"2021-08-14T10:43:36.995628Z","shell.execute_reply":"2021-08-14T10:43:37.007321Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"submission = pd.DataFrame({'id':idx,'label':prob})","metadata":{"execution":{"iopub.status.busy":"2021-08-14T10:43:37.010221Z","iopub.execute_input":"2021-08-14T10:43:37.010552Z","iopub.status.idle":"2021-08-14T10:43:37.043762Z","shell.execute_reply.started":"2021-08-14T10:43:37.010521Z","shell.execute_reply":"2021-08-14T10:43:37.043019Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"submission","metadata":{"execution":{"iopub.status.busy":"2021-08-14T10:43:37.045245Z","iopub.execute_input":"2021-08-14T10:43:37.045736Z","iopub.status.idle":"2021-08-14T10:43:37.092317Z","shell.execute_reply.started":"2021-08-14T10:43:37.045704Z","shell.execute_reply":"2021-08-14T10:43:37.091489Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"submission.to_csv('result.csv',index=False)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T10:43:37.097847Z","iopub.execute_input":"2021-08-14T10:43:37.098431Z","iopub.status.idle":"2021-08-14T10:43:37.519031Z","shell.execute_reply.started":"2021-08-14T10:43:37.098395Z","shell.execute_reply":"2021-08-14T10:43:37.518161Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"### check our model performance and visualize some data!","metadata":{}},{"cell_type":"code","source":"import random\n\nid_list = []\nclass_ = {0: 'cat', 1: 'dog'}\n\nfig, axes = plt.subplots(2, 5, figsize=(20, 12), facecolor='w')\n\nfor ax in axes.ravel():\n    \n    i = random.choice(submission['id'].values)\n    \n    label = submission.loc[submission['id'] == i, 'label'].values[0]\n    if label > 0.5:\n        label = 1\n    else:\n        label = 0\n        \n    img_path = os.path.join(test_dir, '{}.jpg'.format(i))\n    img = Image.open(img_path)\n    \n    ax.set_title(class_[label])\n    ax.imshow(img)","metadata":{"execution":{"iopub.status.busy":"2021-08-14T10:43:37.520615Z","iopub.execute_input":"2021-08-14T10:43:37.521192Z","iopub.status.idle":"2021-08-14T10:43:39.092864Z","shell.execute_reply.started":"2021-08-14T10:43:37.521156Z","shell.execute_reply":"2021-08-14T10:43:39.09182Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]}]}