test_finetune.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import torch\n",
    "import torch.nn as nn\n",
    "import os\n",
    "import numpy as np\n",
    "from model.net import FashionNet\n",
    "from load_dataset import custom_dset, collate_fn, denorm\n",
    "from torch.utils.data import DataLoader\n",
    "import cv2\n",
    "import argparse\n",
    "from utils import AverageMeter, AverageAcc\n",
    "from tqdm import tqdm\n",
    "from focal_loss import *\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<style>\n",
       ".output {\n",
       "    flex-direction: row;\n",
       "}\n",
       "</style>"
      ],
      "text/plain": [
       "<IPython.core.display.HTML object>"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from IPython.display import display, HTML\n",
    "import pandas as pd\n",
    "\n",
    "CSS = \"\"\"\n",
    ".output {\n",
    "    flex-direction: row;\n",
    "}\n",
    "\"\"\"\n",
    "\n",
    "HTML('<style>{}</style>'.format(CSS))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "def test(net, test_loader, accuracy):\n",
    "\n",
    "\n",
    "    losses = AverageMeter()\n",
    "    criterion = FocalLoss(gamma=2, alpha=0.25)\n",
    "    bar = tqdm(total=len(test_loader))\n",
    "\n",
    "    with torch.no_grad():\n",
    "        for it, (imgs, labels, img_ids) in enumerate(test_loader):\n",
    "            bar.update()\n",
    "            imgs = imgs.cuda()\n",
    "            labels = labels.cuda()\n",
    "\n",
    "            # run input through the network\n",
    "            preds = net(imgs)\n",
    "            loss = criterion(preds, labels)\n",
    "\n",
    "\n",
    "            # measure accuracy and record loss\n",
    "            accuracy.update(preds.data, labels.data, topk=(1, 5))\n",
    "            losses.update(loss.item(), imgs.size(0))\n",
    "\n",
    "            if it % 100 == 0:\n",
    "                # save to disk first batch of results for every epoch\n",
    "                for i,im in enumerate(imgs):\n",
    "                    im = im.data.cpu().numpy()\n",
    "                    im = denorm(im.transpose(1,2,0))[...,::-1]\n",
    "\n",
    "                    _, inds = torch.max(preds, 1)\n",
    "                    pred = label_map[inds[i].item()]\n",
    "                    gt_label = label_map[labels[i].item()]\n",
    "\n",
    "                    fname =  str(img_ids[i]) + '__' 'Gt:' + gt_label + 'Pred:' + pred + '.jpg'\n",
    "                    cv2.imwrite(os.path.join(out_path, fname),  im)\n",
    "\n",
    "    bar.close()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "\r",
      "  0%|          | 0/112 [00:00<?, ?it/s]"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Finetune Test Dataset of size: 3135\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "100%|██████████| 112/112 [00:09<00:00, 11.47it/s]\n"
     ]
    }
   ],
   "source": [
    "# check checkpoints folder for model_name\n",
    "model_name = 'finetuned_9.pth'\n",
    "\n",
    "\n",
    "# get the current working directory\n",
    "root = os.getcwd()\n",
    "out_path = './data/show_test'\n",
    "if not os.path.isdir(out_path):\n",
    "    os.mkdir(out_path)\n",
    "\n",
    "\n",
    "# load model\n",
    "model_path = os.path.join(root, 'checkpoints', model_name)\n",
    "model = torch.load(model_path)\n",
    "label_map = model['label_map']\n",
    "test_finetune = custom_dset('test_finetune', label_map)\n",
    "net = FashionNet(len(label_map))\n",
    "net.load_state_dict(model['net'])\n",
    "\n",
    "net = net.cuda()\n",
    "net.eval()\n",
    "\n",
    "\n",
    "label_map = { v:k for k,v in label_map.items()}\n",
    "\n",
    "\n",
    "print('Finetune Test Dataset of size: {}'.format(len(test_finetune)))\n",
    "test_loader = DataLoader(test_finetune, batch_size=28, shuffle=False, collate_fn=collate_fn, num_workers=4)\n",
    "\n",
    "accuracy = AverageAcc(test_finetune.label_map)\n",
    "test(net, test_loader, accuracy)\n",
    "\n",
    "# get accuracies per class and the averages \n",
    "topk_dict, top1, top5 = accuracy()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [],
   "source": [
    "# collect accuracies into lists for printing purpose \n",
    "lc = [k for k in test_finetune.label_map]\n",
    "t1 = [v[1] for k,v in topk_dict.items()]\n",
    "t2 = [v[5] for k,v in topk_dict.items()]\n",
    "avg = [round(top1, 2), round(top5, 2)]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Category</th>\n",
       "      <th>Data examples</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Accessory Gift Set</td>\n",
       "      <td>12</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Bangle</td>\n",
       "      <td>54</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Basketballs</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Bra</td>\n",
       "      <td>131</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Bracelet</td>\n",
       "      <td>33</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Camisoles</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Capris</td>\n",
       "      <td>102</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Caps</td>\n",
       "      <td>131</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Churidar</td>\n",
       "      <td>25</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Clutches</td>\n",
       "      <td>188</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Cufflinks</td>\n",
       "      <td>74</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Dresses</td>\n",
       "      <td>321</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Duffel Bag</td>\n",
       "      <td>40</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Dupatta</td>\n",
       "      <td>98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Earrings</td>\n",
       "      <td>284</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Flats</td>\n",
       "      <td>287</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Footballs</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Free Gifts</td>\n",
       "      <td>66</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Gloves</td>\n",
       "      <td>14</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Headband</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Innerwear Vests</td>\n",
       "      <td>209</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Jackets</td>\n",
       "      <td>67</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Jewellery Set</td>\n",
       "      <td>37</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Key chain</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Kurta Sets</td>\n",
       "      <td>50</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Kurtis</td>\n",
       "      <td>128</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Laptop Bag</td>\n",
       "      <td>36</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Leggings</td>\n",
       "      <td>118</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Lounge Pants</td>\n",
       "      <td>36</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Lounge Shorts</td>\n",
       "      <td>24</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Nightdress</td>\n",
       "      <td>27</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Pendant</td>\n",
       "      <td>83</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Rain Jacket</td>\n",
       "      <td>8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Rain Trousers</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Ring</td>\n",
       "      <td>45</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Robe</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Rucksacks</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Scarves</td>\n",
       "      <td>63</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Shapewear</td>\n",
       "      <td>7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Shoe Accessories</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Shorts</td>\n",
       "      <td>299</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Shrug</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Skirts</td>\n",
       "      <td>65</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Sports Sandals</td>\n",
       "      <td>30</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Stockings</td>\n",
       "      <td>15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Stoles</td>\n",
       "      <td>79</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Sweaters</td>\n",
       "      <td>20</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Sweatshirts</td>\n",
       "      <td>22</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Swimwear</td>\n",
       "      <td>8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Ties</td>\n",
       "      <td>153</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Track Pants</td>\n",
       "      <td>114</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Tracksuits</td>\n",
       "      <td>7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Travel Accessory</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Trousers</td>\n",
       "      <td>237</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Tunics</td>\n",
       "      <td>159</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Umbrellas</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Waist Pouch</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Waistcoat</td>\n",
       "      <td>8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Water Bottle</td>\n",
       "      <td>9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Trainset</th>\n",
       "      <td>Wristbands</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>66 rows × 2 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                    Category Data examples\n",
       "Trainset  Accessory Gift Set            12\n",
       "Trainset              Bangle            54\n",
       "Trainset         Basketballs             2\n",
       "Trainset                 Bra           131\n",
       "Trainset            Bracelet            33\n",
       "Trainset           Camisoles             3\n",
       "Trainset              Capris           102\n",
       "Trainset                Caps           131\n",
       "Trainset            Churidar            25\n",
       "Trainset            Clutches           188\n",
       "Trainset           Cufflinks            74\n",
       "Trainset             Dresses           321\n",
       "Trainset          Duffel Bag            40\n",
       "Trainset             Dupatta            98\n",
       "Trainset            Earrings           284\n",
       "Trainset               Flats           287\n",
       "Trainset           Footballs             6\n",
       "Trainset          Free Gifts            66\n",
       "Trainset              Gloves            14\n",
       "Trainset            Headband             4\n",
       "Trainset     Innerwear Vests           209\n",
       "Trainset             Jackets            67\n",
       "Trainset       Jewellery Set            37\n",
       "Trainset           Key chain             1\n",
       "Trainset          Kurta Sets            50\n",
       "Trainset              Kurtis           128\n",
       "Trainset          Laptop Bag            36\n",
       "Trainset            Leggings           118\n",
       "Trainset        Lounge Pants            36\n",
       "Trainset       Lounge Shorts            24\n",
       "...                      ...           ...\n",
       "Trainset          Nightdress            27\n",
       "Trainset             Pendant            83\n",
       "Trainset         Rain Jacket             8\n",
       "Trainset       Rain Trousers             1\n",
       "Trainset                Ring            45\n",
       "Trainset                Robe             1\n",
       "Trainset           Rucksacks             3\n",
       "Trainset             Scarves            63\n",
       "Trainset           Shapewear             7\n",
       "Trainset    Shoe Accessories             4\n",
       "Trainset              Shorts           299\n",
       "Trainset               Shrug             4\n",
       "Trainset              Skirts            65\n",
       "Trainset      Sports Sandals            30\n",
       "Trainset           Stockings            15\n",
       "Trainset              Stoles            79\n",
       "Trainset            Sweaters            20\n",
       "Trainset         Sweatshirts            22\n",
       "Trainset            Swimwear             8\n",
       "Trainset                Ties           153\n",
       "Trainset         Track Pants           114\n",
       "Trainset          Tracksuits             7\n",
       "Trainset    Travel Accessory             4\n",
       "Trainset            Trousers           237\n",
       "Trainset              Tunics           159\n",
       "Trainset           Umbrellas             1\n",
       "Trainset         Waist Pouch             4\n",
       "Trainset           Waistcoat             8\n",
       "Trainset        Water Bottle             9\n",
       "Trainset          Wristbands             3\n",
       "\n",
       "[66 rows x 2 columns]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Category</th>\n",
       "      <th>Data examples</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Accessory Gift Set</td>\n",
       "      <td>91</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Bangle</td>\n",
       "      <td>20</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Basketballs</td>\n",
       "      <td>11</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Bra</td>\n",
       "      <td>259</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Bracelet</td>\n",
       "      <td>19</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Camisoles</td>\n",
       "      <td>30</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Capris</td>\n",
       "      <td>42</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Caps</td>\n",
       "      <td>110</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Churidar</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Clutches</td>\n",
       "      <td>50</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Cufflinks</td>\n",
       "      <td>29</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Dresses</td>\n",
       "      <td>81</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Duffel Bag</td>\n",
       "      <td>37</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Dupatta</td>\n",
       "      <td>6</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Earrings</td>\n",
       "      <td>71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Flats</td>\n",
       "      <td>138</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Footballs</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Free Gifts</td>\n",
       "      <td>10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Gloves</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Headband</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Innerwear Vests</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Jackets</td>\n",
       "      <td>170</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Jewellery Set</td>\n",
       "      <td>13</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Key chain</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Kurta Sets</td>\n",
       "      <td>23</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Kurtis</td>\n",
       "      <td>78</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Laptop Bag</td>\n",
       "      <td>32</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Leggings</td>\n",
       "      <td>40</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Lounge Pants</td>\n",
       "      <td>9</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Lounge Shorts</td>\n",
       "      <td>8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Nightdress</td>\n",
       "      <td>143</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Pendant</td>\n",
       "      <td>69</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Rain Jacket</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Rain Trousers</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Ring</td>\n",
       "      <td>43</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Robe</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Rucksacks</td>\n",
       "      <td>8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Scarves</td>\n",
       "      <td>41</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Shapewear</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Shoe Accessories</td>\n",
       "      <td>15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Shorts</td>\n",
       "      <td>181</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Shrug</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Skirts</td>\n",
       "      <td>45</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Sports Sandals</td>\n",
       "      <td>25</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Stockings</td>\n",
       "      <td>13</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Stoles</td>\n",
       "      <td>10</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Sweaters</td>\n",
       "      <td>235</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Sweatshirts</td>\n",
       "      <td>223</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Swimwear</td>\n",
       "      <td>7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Ties</td>\n",
       "      <td>77</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Track Pants</td>\n",
       "      <td>142</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Tracksuits</td>\n",
       "      <td>16</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Travel Accessory</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Trousers</td>\n",
       "      <td>219</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Tunics</td>\n",
       "      <td>36</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Umbrellas</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Waist Pouch</td>\n",
       "      <td>13</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Waistcoat</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Water Bottle</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Wristbands</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>66 rows × 2 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                   Category Data examples\n",
       "Testset  Accessory Gift Set            91\n",
       "Testset              Bangle            20\n",
       "Testset         Basketballs            11\n",
       "Testset                 Bra           259\n",
       "Testset            Bracelet            19\n",
       "Testset           Camisoles            30\n",
       "Testset              Capris            42\n",
       "Testset                Caps           110\n",
       "Testset            Churidar             4\n",
       "Testset            Clutches            50\n",
       "Testset           Cufflinks            29\n",
       "Testset             Dresses            81\n",
       "Testset          Duffel Bag            37\n",
       "Testset             Dupatta             6\n",
       "Testset            Earrings            71\n",
       "Testset               Flats           138\n",
       "Testset           Footballs             1\n",
       "Testset          Free Gifts            10\n",
       "Testset              Gloves             5\n",
       "Testset            Headband             1\n",
       "Testset     Innerwear Vests             1\n",
       "Testset             Jackets           170\n",
       "Testset       Jewellery Set            13\n",
       "Testset           Key chain             1\n",
       "Testset          Kurta Sets            23\n",
       "Testset              Kurtis            78\n",
       "Testset          Laptop Bag            32\n",
       "Testset            Leggings            40\n",
       "Testset        Lounge Pants             9\n",
       "Testset       Lounge Shorts             8\n",
       "...                     ...           ...\n",
       "Testset          Nightdress           143\n",
       "Testset             Pendant            69\n",
       "Testset         Rain Jacket             2\n",
       "Testset       Rain Trousers             1\n",
       "Testset                Ring            43\n",
       "Testset                Robe             2\n",
       "Testset           Rucksacks             8\n",
       "Testset             Scarves            41\n",
       "Testset           Shapewear             1\n",
       "Testset    Shoe Accessories            15\n",
       "Testset              Shorts           181\n",
       "Testset               Shrug             1\n",
       "Testset              Skirts            45\n",
       "Testset      Sports Sandals            25\n",
       "Testset           Stockings            13\n",
       "Testset              Stoles            10\n",
       "Testset            Sweaters           235\n",
       "Testset         Sweatshirts           223\n",
       "Testset            Swimwear             7\n",
       "Testset                Ties            77\n",
       "Testset         Track Pants           142\n",
       "Testset          Tracksuits            16\n",
       "Testset    Travel Accessory             5\n",
       "Testset            Trousers           219\n",
       "Testset              Tunics            36\n",
       "Testset           Umbrellas             5\n",
       "Testset         Waist Pouch            13\n",
       "Testset           Waistcoat             4\n",
       "Testset        Water Bottle             1\n",
       "Testset          Wristbands             3\n",
       "\n",
       "[66 rows x 2 columns]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Category</th>\n",
       "      <th>Top1</th>\n",
       "      <th>Top5</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Accessory Gift Set</td>\n",
       "      <td>0.95</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bangle</td>\n",
       "      <td>0.25</td>\n",
       "      <td>0.85</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Basketballs</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bra</td>\n",
       "      <td>0.95</td>\n",
       "      <td>0.98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bracelet</td>\n",
       "      <td>0.53</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Camisoles</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Capris</td>\n",
       "      <td>0.6</td>\n",
       "      <td>0.98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Caps</td>\n",
       "      <td>0.7</td>\n",
       "      <td>0.85</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Churidar</td>\n",
       "      <td>0.25</td>\n",
       "      <td>0.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Clutches</td>\n",
       "      <td>0.96</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Cufflinks</td>\n",
       "      <td>0.9</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Dresses</td>\n",
       "      <td>0.74</td>\n",
       "      <td>0.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Duffel Bag</td>\n",
       "      <td>0.78</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Dupatta</td>\n",
       "      <td>0.17</td>\n",
       "      <td>0.83</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Earrings</td>\n",
       "      <td>0.99</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Flats</td>\n",
       "      <td>0.88</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Footballs</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Free Gifts</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Gloves</td>\n",
       "      <td>0.4</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Headband</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Innerwear Vests</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Jackets</td>\n",
       "      <td>0.16</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Jewellery Set</td>\n",
       "      <td>0.69</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Key chain</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Kurta Sets</td>\n",
       "      <td>0.7</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Kurtis</td>\n",
       "      <td>0.47</td>\n",
       "      <td>0.99</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Laptop Bag</td>\n",
       "      <td>0.53</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Leggings</td>\n",
       "      <td>0.28</td>\n",
       "      <td>0.85</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Lounge Pants</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Lounge Shorts</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.38</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Nightdress</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.38</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Pendant</td>\n",
       "      <td>0.7</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Jacket</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Trousers</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ring</td>\n",
       "      <td>0.91</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Robe</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rucksacks</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.38</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Scarves</td>\n",
       "      <td>0.68</td>\n",
       "      <td>0.83</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shapewear</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shoe Accessories</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.07</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shorts</td>\n",
       "      <td>0.78</td>\n",
       "      <td>0.97</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shrug</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Skirts</td>\n",
       "      <td>0.42</td>\n",
       "      <td>0.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sports Sandals</td>\n",
       "      <td>0.68</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stockings</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.46</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stoles</td>\n",
       "      <td>0.2</td>\n",
       "      <td>0.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweaters</td>\n",
       "      <td>0.23</td>\n",
       "      <td>0.79</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweatshirts</td>\n",
       "      <td>0.82</td>\n",
       "      <td>0.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Swimwear</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ties</td>\n",
       "      <td>0.97</td>\n",
       "      <td>0.99</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Track Pants</td>\n",
       "      <td>0.75</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tracksuits</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.69</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Travel Accessory</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Trousers</td>\n",
       "      <td>0.78</td>\n",
       "      <td>0.98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tunics</td>\n",
       "      <td>0.28</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Umbrellas</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waist Pouch</td>\n",
       "      <td>0.08</td>\n",
       "      <td>0.54</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waistcoat</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.25</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Water Bottle</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Wristbands</td>\n",
       "      <td>0.67</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>66 rows × 3 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                   Category  Top1  Top5\n",
       "TestAcc  Accessory Gift Set  0.95   1.0\n",
       "TestAcc              Bangle  0.25  0.85\n",
       "TestAcc         Basketballs   0.0   0.0\n",
       "TestAcc                 Bra  0.95  0.98\n",
       "TestAcc            Bracelet  0.53   1.0\n",
       "TestAcc           Camisoles   0.0   0.2\n",
       "TestAcc              Capris   0.6  0.98\n",
       "TestAcc                Caps   0.7  0.85\n",
       "TestAcc            Churidar  0.25   0.5\n",
       "TestAcc            Clutches  0.96   1.0\n",
       "TestAcc           Cufflinks   0.9   1.0\n",
       "TestAcc             Dresses  0.74  0.93\n",
       "TestAcc          Duffel Bag  0.78   1.0\n",
       "TestAcc             Dupatta  0.17  0.83\n",
       "TestAcc            Earrings  0.99   1.0\n",
       "TestAcc               Flats  0.88   1.0\n",
       "TestAcc           Footballs   1.0   1.0\n",
       "TestAcc          Free Gifts   0.0   0.2\n",
       "TestAcc              Gloves   0.4   1.0\n",
       "TestAcc            Headband   1.0   1.0\n",
       "TestAcc     Innerwear Vests   1.0   1.0\n",
       "TestAcc             Jackets  0.16  0.94\n",
       "TestAcc       Jewellery Set  0.69   1.0\n",
       "TestAcc           Key chain   0.0   0.0\n",
       "TestAcc          Kurta Sets   0.7   1.0\n",
       "TestAcc              Kurtis  0.47  0.99\n",
       "TestAcc          Laptop Bag  0.53  0.94\n",
       "TestAcc            Leggings  0.28  0.85\n",
       "TestAcc        Lounge Pants   0.0   0.0\n",
       "TestAcc       Lounge Shorts   0.0  0.38\n",
       "...                     ...   ...   ...\n",
       "TestAcc          Nightdress  0.01  0.38\n",
       "TestAcc             Pendant   0.7  0.88\n",
       "TestAcc         Rain Jacket   0.0   1.0\n",
       "TestAcc       Rain Trousers   0.0   1.0\n",
       "TestAcc                Ring  0.91  0.95\n",
       "TestAcc                Robe   0.0   0.0\n",
       "TestAcc           Rucksacks   0.0  0.38\n",
       "TestAcc             Scarves  0.68  0.83\n",
       "TestAcc           Shapewear   0.0   0.0\n",
       "TestAcc    Shoe Accessories   0.0  0.07\n",
       "TestAcc              Shorts  0.78  0.97\n",
       "TestAcc               Shrug   0.0   0.0\n",
       "TestAcc              Skirts  0.42  0.71\n",
       "TestAcc      Sports Sandals  0.68   1.0\n",
       "TestAcc           Stockings  0.15  0.46\n",
       "TestAcc              Stoles   0.2   0.7\n",
       "TestAcc            Sweaters  0.23  0.79\n",
       "TestAcc         Sweatshirts  0.82  0.93\n",
       "TestAcc            Swimwear   0.0   0.0\n",
       "TestAcc                Ties  0.97  0.99\n",
       "TestAcc         Track Pants  0.75  0.95\n",
       "TestAcc          Tracksuits   0.0  0.69\n",
       "TestAcc    Travel Accessory   0.0   0.2\n",
       "TestAcc            Trousers  0.78  0.98\n",
       "TestAcc              Tunics  0.28  0.94\n",
       "TestAcc           Umbrellas   1.0   1.0\n",
       "TestAcc         Waist Pouch  0.08  0.54\n",
       "TestAcc           Waistcoat   0.0  0.25\n",
       "TestAcc        Water Bottle   0.0   1.0\n",
       "TestAcc          Wristbands  0.67   1.0\n",
       "\n",
       "[66 rows x 3 columns]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "trainset =  custom_dset('train_finetune')\n",
    "lc_train = [trainset.freq[c] for c in lc]\n",
    "\n",
    "# display train set statitstics \n",
    "df_dset = pd.DataFrame(np.vstack((lc, lc_train)).transpose(), index=['Trainset']*66, columns=['Category', 'Data examples'])\n",
    "display(df_dset)\n",
    "\n",
    "# display test set statitstics \n",
    "df_dset = pd.DataFrame(np.vstack( (lc, [v[1] for _,v in accuracy.count.items()])).transpose(), index=['Testset']*66, columns=['Category', 'Data examples'])\n",
    "display(df_dset)\n",
    "\n",
    "# concatenate accuracy lists in numpy array of shape (num_classes, num_columns)\n",
    "df_class = pd.DataFrame(np.vstack( (lc, t1, t2)).transpose(), columns=['Category', 'Top1', 'Top5'], index=['TestAcc']*66)\n",
    "df_avg = pd.DataFrame(np.array(avg).reshape(1,2), columns=['Top1', 'Top5'], index=['Average'])\n",
    "\n",
    "# display the accuracies \n",
    "display(df_class)\n",
    "#display(df_avg)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Category</th>\n",
       "      <th>Top1</th>\n",
       "      <th>Top5</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Accessory Gift Set</td>\n",
       "      <td>0.95</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bangle</td>\n",
       "      <td>0.25</td>\n",
       "      <td>0.85</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Basketballs</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bra</td>\n",
       "      <td>0.95</td>\n",
       "      <td>0.98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bracelet</td>\n",
       "      <td>0.53</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Camisoles</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Capris</td>\n",
       "      <td>0.6</td>\n",
       "      <td>0.98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Caps</td>\n",
       "      <td>0.7</td>\n",
       "      <td>0.85</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Churidar</td>\n",
       "      <td>0.25</td>\n",
       "      <td>0.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Clutches</td>\n",
       "      <td>0.96</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Cufflinks</td>\n",
       "      <td>0.9</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Dresses</td>\n",
       "      <td>0.74</td>\n",
       "      <td>0.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Duffel Bag</td>\n",
       "      <td>0.78</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Dupatta</td>\n",
       "      <td>0.17</td>\n",
       "      <td>0.83</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Earrings</td>\n",
       "      <td>0.99</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Flats</td>\n",
       "      <td>0.88</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Footballs</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Free Gifts</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Gloves</td>\n",
       "      <td>0.4</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Headband</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Innerwear Vests</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Jackets</td>\n",
       "      <td>0.16</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Jewellery Set</td>\n",
       "      <td>0.69</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Key chain</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Kurta Sets</td>\n",
       "      <td>0.7</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Kurtis</td>\n",
       "      <td>0.47</td>\n",
       "      <td>0.99</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Laptop Bag</td>\n",
       "      <td>0.53</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Leggings</td>\n",
       "      <td>0.28</td>\n",
       "      <td>0.85</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Lounge Pants</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Lounge Shorts</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.38</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Nightdress</td>\n",
       "      <td>0.01</td>\n",
       "      <td>0.38</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Pendant</td>\n",
       "      <td>0.7</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Jacket</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Trousers</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ring</td>\n",
       "      <td>0.91</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Robe</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rucksacks</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.38</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Scarves</td>\n",
       "      <td>0.68</td>\n",
       "      <td>0.83</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shapewear</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shoe Accessories</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.07</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shorts</td>\n",
       "      <td>0.78</td>\n",
       "      <td>0.97</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shrug</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Skirts</td>\n",
       "      <td>0.42</td>\n",
       "      <td>0.71</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sports Sandals</td>\n",
       "      <td>0.68</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stockings</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.46</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stoles</td>\n",
       "      <td>0.2</td>\n",
       "      <td>0.7</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweaters</td>\n",
       "      <td>0.23</td>\n",
       "      <td>0.79</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweatshirts</td>\n",
       "      <td>0.82</td>\n",
       "      <td>0.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Swimwear</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ties</td>\n",
       "      <td>0.97</td>\n",
       "      <td>0.99</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Track Pants</td>\n",
       "      <td>0.75</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tracksuits</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.69</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Travel Accessory</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Trousers</td>\n",
       "      <td>0.78</td>\n",
       "      <td>0.98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tunics</td>\n",
       "      <td>0.28</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Umbrellas</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waist Pouch</td>\n",
       "      <td>0.08</td>\n",
       "      <td>0.54</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waistcoat</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.25</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Water Bottle</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Wristbands</td>\n",
       "      <td>0.67</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>66 rows × 3 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                   Category  Top1  Top5\n",
       "TestAcc  Accessory Gift Set  0.95   1.0\n",
       "TestAcc              Bangle  0.25  0.85\n",
       "TestAcc         Basketballs   0.0   0.0\n",
       "TestAcc                 Bra  0.95  0.98\n",
       "TestAcc            Bracelet  0.53   1.0\n",
       "TestAcc           Camisoles   0.0   0.2\n",
       "TestAcc              Capris   0.6  0.98\n",
       "TestAcc                Caps   0.7  0.85\n",
       "TestAcc            Churidar  0.25   0.5\n",
       "TestAcc            Clutches  0.96   1.0\n",
       "TestAcc           Cufflinks   0.9   1.0\n",
       "TestAcc             Dresses  0.74  0.93\n",
       "TestAcc          Duffel Bag  0.78   1.0\n",
       "TestAcc             Dupatta  0.17  0.83\n",
       "TestAcc            Earrings  0.99   1.0\n",
       "TestAcc               Flats  0.88   1.0\n",
       "TestAcc           Footballs   1.0   1.0\n",
       "TestAcc          Free Gifts   0.0   0.2\n",
       "TestAcc              Gloves   0.4   1.0\n",
       "TestAcc            Headband   1.0   1.0\n",
       "TestAcc     Innerwear Vests   1.0   1.0\n",
       "TestAcc             Jackets  0.16  0.94\n",
       "TestAcc       Jewellery Set  0.69   1.0\n",
       "TestAcc           Key chain   0.0   0.0\n",
       "TestAcc          Kurta Sets   0.7   1.0\n",
       "TestAcc              Kurtis  0.47  0.99\n",
       "TestAcc          Laptop Bag  0.53  0.94\n",
       "TestAcc            Leggings  0.28  0.85\n",
       "TestAcc        Lounge Pants   0.0   0.0\n",
       "TestAcc       Lounge Shorts   0.0  0.38\n",
       "...                     ...   ...   ...\n",
       "TestAcc          Nightdress  0.01  0.38\n",
       "TestAcc             Pendant   0.7  0.88\n",
       "TestAcc         Rain Jacket   0.0   1.0\n",
       "TestAcc       Rain Trousers   0.0   1.0\n",
       "TestAcc                Ring  0.91  0.95\n",
       "TestAcc                Robe   0.0   0.0\n",
       "TestAcc           Rucksacks   0.0  0.38\n",
       "TestAcc             Scarves  0.68  0.83\n",
       "TestAcc           Shapewear   0.0   0.0\n",
       "TestAcc    Shoe Accessories   0.0  0.07\n",
       "TestAcc              Shorts  0.78  0.97\n",
       "TestAcc               Shrug   0.0   0.0\n",
       "TestAcc              Skirts  0.42  0.71\n",
       "TestAcc      Sports Sandals  0.68   1.0\n",
       "TestAcc           Stockings  0.15  0.46\n",
       "TestAcc              Stoles   0.2   0.7\n",
       "TestAcc            Sweaters  0.23  0.79\n",
       "TestAcc         Sweatshirts  0.82  0.93\n",
       "TestAcc            Swimwear   0.0   0.0\n",
       "TestAcc                Ties  0.97  0.99\n",
       "TestAcc         Track Pants  0.75  0.95\n",
       "TestAcc          Tracksuits   0.0  0.69\n",
       "TestAcc    Travel Accessory   0.0   0.2\n",
       "TestAcc            Trousers  0.78  0.98\n",
       "TestAcc              Tunics  0.28  0.94\n",
       "TestAcc           Umbrellas   1.0   1.0\n",
       "TestAcc         Waist Pouch  0.08  0.54\n",
       "TestAcc           Waistcoat   0.0  0.25\n",
       "TestAcc        Water Bottle   0.0   1.0\n",
       "TestAcc          Wristbands  0.67   1.0\n",
       "\n",
       "[66 rows x 3 columns]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Average Top1</th>\n",
       "      <th>Average Top5</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>training</th>\n",
       "      <td>0.42</td>\n",
       "      <td>0.71</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "          Average Top1  Average Top5\n",
       "training          0.42          0.71"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "\n",
    "\n",
    "# concatenate accuracy lists in numpy array of shape (num_classes, num_columns)\n",
    "df_class = pd.DataFrame(np.vstack( (lc, t1, t2)).transpose(), columns=['Category', 'Top1', 'Top5'], index=['TestAcc']*66)\n",
    "df_avg = pd.DataFrame(np.array(avg).reshape(1,2), columns=['Average Top1', 'Average Top5'], index=['training'])\n",
    "\n",
    "# display the accuracies \n",
    "display(df_class)\n",
    "display(df_avg)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "sorted(test_finetune.label_map, key=str.lower)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Category</th>\n",
       "      <th>Top1</th>\n",
       "      <th>Top5</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Accessory Gift Set</td>\n",
       "      <td>0.98</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bangle</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Basketballs</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bra</td>\n",
       "      <td>0.95</td>\n",
       "      <td>0.97</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Bracelet</td>\n",
       "      <td>0.63</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Camisoles</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Capris</td>\n",
       "      <td>0.33</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Caps</td>\n",
       "      <td>0.79</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Churidar</td>\n",
       "      <td>0.5</td>\n",
       "      <td>0.75</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Clutches</td>\n",
       "      <td>0.96</td>\n",
       "      <td>0.98</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Cufflinks</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Dresses</td>\n",
       "      <td>0.54</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Duffel Bag</td>\n",
       "      <td>0.54</td>\n",
       "      <td>0.92</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Dupatta</td>\n",
       "      <td>0.17</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Earrings</td>\n",
       "      <td>0.96</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Flats</td>\n",
       "      <td>0.96</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Footballs</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Free Gifts</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Gloves</td>\n",
       "      <td>0.4</td>\n",
       "      <td>0.8</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Headband</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Innerwear Vests</td>\n",
       "      <td>1.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Jackets</td>\n",
       "      <td>0.44</td>\n",
       "      <td>0.85</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Jewellery Set</td>\n",
       "      <td>0.54</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Key chain</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Kurta Sets</td>\n",
       "      <td>0.87</td>\n",
       "      <td>0.96</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Kurtis</td>\n",
       "      <td>0.64</td>\n",
       "      <td>0.97</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Laptop Bag</td>\n",
       "      <td>0.91</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Leggings</td>\n",
       "      <td>0.15</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Lounge Pants</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.56</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Lounge Shorts</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.38</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Nightdress</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.52</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Pendant</td>\n",
       "      <td>0.87</td>\n",
       "      <td>0.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Jacket</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Trousers</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ring</td>\n",
       "      <td>0.86</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Robe</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rucksacks</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.12</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Scarves</td>\n",
       "      <td>0.27</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shapewear</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shoe Accessories</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.07</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shorts</td>\n",
       "      <td>0.88</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shrug</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Skirts</td>\n",
       "      <td>0.13</td>\n",
       "      <td>0.76</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sports Sandals</td>\n",
       "      <td>0.68</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stockings</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.23</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stoles</td>\n",
       "      <td>0.1</td>\n",
       "      <td>0.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweaters</td>\n",
       "      <td>0.07</td>\n",
       "      <td>0.78</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweatshirts</td>\n",
       "      <td>0.35</td>\n",
       "      <td>0.84</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Swimwear</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.29</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ties</td>\n",
       "      <td>0.94</td>\n",
       "      <td>0.99</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Track Pants</td>\n",
       "      <td>0.87</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tracksuits</td>\n",
       "      <td>0.12</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Travel Accessory</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Trousers</td>\n",
       "      <td>0.67</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tunics</td>\n",
       "      <td>0.28</td>\n",
       "      <td>0.89</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Umbrellas</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waist Pouch</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waistcoat</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.75</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Water Bottle</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Wristbands</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.33</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>66 rows × 3 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                   Category  Top1  Top5\n",
       "TestAcc  Accessory Gift Set  0.98   1.0\n",
       "TestAcc              Bangle  0.05   0.8\n",
       "TestAcc         Basketballs   0.0   0.0\n",
       "TestAcc                 Bra  0.95  0.97\n",
       "TestAcc            Bracelet  0.63   1.0\n",
       "TestAcc           Camisoles   0.0   0.0\n",
       "TestAcc              Capris  0.33  0.95\n",
       "TestAcc                Caps  0.79  0.88\n",
       "TestAcc            Churidar   0.5  0.75\n",
       "TestAcc            Clutches  0.96  0.98\n",
       "TestAcc           Cufflinks   1.0   1.0\n",
       "TestAcc             Dresses  0.54  0.95\n",
       "TestAcc          Duffel Bag  0.54  0.92\n",
       "TestAcc             Dupatta  0.17   1.0\n",
       "TestAcc            Earrings  0.96   1.0\n",
       "TestAcc               Flats  0.96   1.0\n",
       "TestAcc           Footballs   0.0   1.0\n",
       "TestAcc          Free Gifts   0.0   0.0\n",
       "TestAcc              Gloves   0.4   0.8\n",
       "TestAcc            Headband   0.0   0.0\n",
       "TestAcc     Innerwear Vests   1.0   1.0\n",
       "TestAcc             Jackets  0.44  0.85\n",
       "TestAcc       Jewellery Set  0.54   1.0\n",
       "TestAcc           Key chain   0.0   0.0\n",
       "TestAcc          Kurta Sets  0.87  0.96\n",
       "TestAcc              Kurtis  0.64  0.97\n",
       "TestAcc          Laptop Bag  0.91   1.0\n",
       "TestAcc            Leggings  0.15  0.88\n",
       "TestAcc        Lounge Pants   0.0  0.56\n",
       "TestAcc       Lounge Shorts   0.0  0.38\n",
       "...                     ...   ...   ...\n",
       "TestAcc          Nightdress  0.05  0.52\n",
       "TestAcc             Pendant  0.87  0.93\n",
       "TestAcc         Rain Jacket   0.0   0.5\n",
       "TestAcc       Rain Trousers   0.0   0.0\n",
       "TestAcc                Ring  0.86  0.95\n",
       "TestAcc                Robe   0.0   0.0\n",
       "TestAcc           Rucksacks   0.0  0.12\n",
       "TestAcc             Scarves  0.27  0.88\n",
       "TestAcc           Shapewear   0.0   0.0\n",
       "TestAcc    Shoe Accessories   0.0  0.07\n",
       "TestAcc              Shorts  0.88  0.94\n",
       "TestAcc               Shrug   0.0   0.0\n",
       "TestAcc              Skirts  0.13  0.76\n",
       "TestAcc      Sports Sandals  0.68   1.0\n",
       "TestAcc           Stockings   0.0  0.23\n",
       "TestAcc              Stoles   0.1   0.5\n",
       "TestAcc            Sweaters  0.07  0.78\n",
       "TestAcc         Sweatshirts  0.35  0.84\n",
       "TestAcc            Swimwear   0.0  0.29\n",
       "TestAcc                Ties  0.94  0.99\n",
       "TestAcc         Track Pants  0.87  0.94\n",
       "TestAcc          Tracksuits  0.12  0.88\n",
       "TestAcc    Travel Accessory   0.0   0.2\n",
       "TestAcc            Trousers  0.67  0.94\n",
       "TestAcc              Tunics  0.28  0.89\n",
       "TestAcc           Umbrellas   0.0   0.0\n",
       "TestAcc         Waist Pouch   0.0  0.15\n",
       "TestAcc           Waistcoat   0.0  0.75\n",
       "TestAcc        Water Bottle   0.0   1.0\n",
       "TestAcc          Wristbands   0.0  0.33\n",
       "\n",
       "[66 rows x 3 columns]"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df_class\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>Category</th>\n",
       "      <th>Data examples</th>\n",
       "      <th>Top1</th>\n",
       "      <th>Top5</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Accessory Gift Set</td>\n",
       "      <td>91</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Bangle</td>\n",
       "      <td>20</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Basketballs</td>\n",
       "      <td>11</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Bra</td>\n",
       "      <td>259</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Bracelet</td>\n",
       "      <td>19</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Camisoles</td>\n",
       "      <td>30</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Capris</td>\n",
       "      <td>42</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Caps</td>\n",
       "      <td>110</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Churidar</td>\n",
       "      <td>4</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Clutches</td>\n",
       "      <td>50</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Cufflinks</td>\n",
       "      <td>29</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Dresses</td>\n",
       "      <td>81</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Duffel Bag</td>\n",
       "      <td>37</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Dupatta</td>\n",
       "      <td>6</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Earrings</td>\n",
       "      <td>71</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Flats</td>\n",
       "      <td>138</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Footballs</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Free Gifts</td>\n",
       "      <td>10</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Gloves</td>\n",
       "      <td>5</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Headband</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Innerwear Vests</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Jackets</td>\n",
       "      <td>170</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Jewellery Set</td>\n",
       "      <td>13</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Key chain</td>\n",
       "      <td>1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Kurta Sets</td>\n",
       "      <td>23</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Kurtis</td>\n",
       "      <td>78</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Laptop Bag</td>\n",
       "      <td>32</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Leggings</td>\n",
       "      <td>40</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Lounge Pants</td>\n",
       "      <td>9</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>Testset</th>\n",
       "      <td>Lounge Shorts</td>\n",
       "      <td>8</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Nightdress</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.05</td>\n",
       "      <td>0.52</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Pendant</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.87</td>\n",
       "      <td>0.93</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Jacket</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rain Trousers</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ring</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.86</td>\n",
       "      <td>0.95</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Robe</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Rucksacks</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.12</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Scarves</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.27</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shapewear</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shoe Accessories</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.07</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shorts</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.88</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Shrug</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Skirts</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.13</td>\n",
       "      <td>0.76</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sports Sandals</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.68</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stockings</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.23</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Stoles</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.1</td>\n",
       "      <td>0.5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweaters</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.07</td>\n",
       "      <td>0.78</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Sweatshirts</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.35</td>\n",
       "      <td>0.84</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Swimwear</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.29</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Ties</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.94</td>\n",
       "      <td>0.99</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Track Pants</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.87</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tracksuits</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.12</td>\n",
       "      <td>0.88</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Travel Accessory</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Trousers</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.67</td>\n",
       "      <td>0.94</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Tunics</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.28</td>\n",
       "      <td>0.89</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Umbrellas</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waist Pouch</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.15</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Waistcoat</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.75</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Water Bottle</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>1.0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>TestAcc</th>\n",
       "      <td>Wristbands</td>\n",
       "      <td>NaN</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.33</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>132 rows × 4 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                   Category Data examples  Top1  Top5\n",
       "Testset  Accessory Gift Set            91   NaN   NaN\n",
       "Testset              Bangle            20   NaN   NaN\n",
       "Testset         Basketballs            11   NaN   NaN\n",
       "Testset                 Bra           259   NaN   NaN\n",
       "Testset            Bracelet            19   NaN   NaN\n",
       "Testset           Camisoles            30   NaN   NaN\n",
       "Testset              Capris            42   NaN   NaN\n",
       "Testset                Caps           110   NaN   NaN\n",
       "Testset            Churidar             4   NaN   NaN\n",
       "Testset            Clutches            50   NaN   NaN\n",
       "Testset           Cufflinks            29   NaN   NaN\n",
       "Testset             Dresses            81   NaN   NaN\n",
       "Testset          Duffel Bag            37   NaN   NaN\n",
       "Testset             Dupatta             6   NaN   NaN\n",
       "Testset            Earrings            71   NaN   NaN\n",
       "Testset               Flats           138   NaN   NaN\n",
       "Testset           Footballs             1   NaN   NaN\n",
       "Testset          Free Gifts            10   NaN   NaN\n",
       "Testset              Gloves             5   NaN   NaN\n",
       "Testset            Headband             1   NaN   NaN\n",
       "Testset     Innerwear Vests             1   NaN   NaN\n",
       "Testset             Jackets           170   NaN   NaN\n",
       "Testset       Jewellery Set            13   NaN   NaN\n",
       "Testset           Key chain             1   NaN   NaN\n",
       "Testset          Kurta Sets            23   NaN   NaN\n",
       "Testset              Kurtis            78   NaN   NaN\n",
       "Testset          Laptop Bag            32   NaN   NaN\n",
       "Testset            Leggings            40   NaN   NaN\n",
       "Testset        Lounge Pants             9   NaN   NaN\n",
       "Testset       Lounge Shorts             8   NaN   NaN\n",
       "...                     ...           ...   ...   ...\n",
       "TestAcc          Nightdress           NaN  0.05  0.52\n",
       "TestAcc             Pendant           NaN  0.87  0.93\n",
       "TestAcc         Rain Jacket           NaN   0.0   0.5\n",
       "TestAcc       Rain Trousers           NaN   0.0   0.0\n",
       "TestAcc                Ring           NaN  0.86  0.95\n",
       "TestAcc                Robe           NaN   0.0   0.0\n",
       "TestAcc           Rucksacks           NaN   0.0  0.12\n",
       "TestAcc             Scarves           NaN  0.27  0.88\n",
       "TestAcc           Shapewear           NaN   0.0   0.0\n",
       "TestAcc    Shoe Accessories           NaN   0.0  0.07\n",
       "TestAcc              Shorts           NaN  0.88  0.94\n",
       "TestAcc               Shrug           NaN   0.0   0.0\n",
       "TestAcc              Skirts           NaN  0.13  0.76\n",
       "TestAcc      Sports Sandals           NaN  0.68   1.0\n",
       "TestAcc           Stockings           NaN   0.0  0.23\n",
       "TestAcc              Stoles           NaN   0.1   0.5\n",
       "TestAcc            Sweaters           NaN  0.07  0.78\n",
       "TestAcc         Sweatshirts           NaN  0.35  0.84\n",
       "TestAcc            Swimwear           NaN   0.0  0.29\n",
       "TestAcc                Ties           NaN  0.94  0.99\n",
       "TestAcc         Track Pants           NaN  0.87  0.94\n",
       "TestAcc          Tracksuits           NaN  0.12  0.88\n",
       "TestAcc    Travel Accessory           NaN   0.0   0.2\n",
       "TestAcc            Trousers           NaN  0.67  0.94\n",
       "TestAcc              Tunics           NaN  0.28  0.89\n",
       "TestAcc           Umbrellas           NaN   0.0   0.0\n",
       "TestAcc         Waist Pouch           NaN   0.0  0.15\n",
       "TestAcc           Waistcoat           NaN   0.0  0.75\n",
       "TestAcc        Water Bottle           NaN   0.0   1.0\n",
       "TestAcc          Wristbands           NaN   0.0  0.33\n",
       "\n",
       "[132 rows x 4 columns]"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "pd.concat([df_dset, df_class], ignore_index=False, sort=False, join='outer')\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}