20221029_code.ipynb

{
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/ZzangJay/team_project/blob/main/20221029_code.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 1,
      "metadata": {
        "id": "xCBwZEG88iTZ",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "f2123a16-408f-4eb1-f2bc-741cb58980cf"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "/usr/local/lib/python3.7/dist-packages/ipykernel_launcher.py:9: TqdmExperimentalWarning: Using `tqdm.autonotebook.tqdm` in notebook mode. Use `tqdm.tqdm` instead to force console mode (e.g. in jupyter console)\n",
            "  if __name__ == '__main__':\n"
          ]
        }
      ],
      "source": [
        "import pandas as pd\n",
        "import numpy as np\n",
        "#import tensorflow as tf \n",
        "import os\n",
        "import sys\n",
        "from datetime import datetime, timedelta\n",
        "\n",
        "\n",
        "from tqdm.autonotebook import tqdm\n",
        "import csv\n",
        "\n",
        "import re\n",
        "\n",
        "import gzip\n",
        "import shutil"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 2,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "dKfa-fAw8qIn",
        "outputId": "3e046e0f-51f4-41c5-bcd7-d6ffdd3bfea9"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Mounted at /content/gdrive/\n"
          ]
        },
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "['CHANGELOG.txt',\n",
              " 'LICENSE.txt',\n",
              " 'SHA256SUMS.txt',\n",
              " 'index.html',\n",
              " 'icu',\n",
              " 'hosp',\n",
              " 'drop_2_chartevents.csv']"
            ]
          },
          "metadata": {},
          "execution_count": 2
        }
      ],
      "source": [
        "from google.colab import drive  # for colab\n",
        "drive.mount('/content/gdrive/')  # for colab\n",
        "cloud_directory = '/content/gdrive/MyDrive/mimiciv_unzip/'\n",
        "os.listdir(cloud_directory) # for colab, check cloud directory mount"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "d_items = pd.read_csv('/content/gdrive/MyDrive/mimiciv_unzip/icu/d_items.csv', low_memory=False, dtype='str', keep_default_na=False)"
      ],
      "metadata": {
        "id": "Y9xRZWTbsBDh"
      },
      "execution_count": 3,
      "outputs": []
    },
    {
      "cell_type": "code",
      "execution_count": 4,
      "metadata": {
        "id": "gicdKWPb8xfP"
      },
      "outputs": [],
      "source": [
        "admission = pd.read_csv('/content/gdrive/MyDrive/mimiciv_unzip/hosp/admissions.csv', low_memory=False, dtype='str', keep_default_na=False)\n",
        "patients = pd.read_csv('/content/gdrive/MyDrive/mimiciv_unzip/hosp/patients.csv', low_memory=False, dtype='str', keep_default_na=False)\n",
        "diagnoses_icd = pd.read_csv('/content/gdrive/MyDrive/mimiciv_unzip/hosp/diagnoses_icd.csv', low_memory=False, dtype='str', keep_default_na=False)\n",
        "icustays = pd.read_csv('/content/gdrive/MyDrive/mimiciv_unzip/icu/icustays.csv', low_memory=False, dtype='str', keep_default_na=False)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 67,
      "metadata": {
        "id": "pCjNWoWU9HQ9"
      },
      "outputs": [],
      "source": [
        "sepsis = pd.read_csv('/content/gdrive/MyDrive/mimiciv_der/sepsis3.csv', low_memory=False, dtype='str', keep_default_na=False)"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 70,
      "metadata": {
        "id": "M4GF5YNSEXLE"
      },
      "outputs": [],
      "source": [
        "admission = pd.merge(left = admission, right = sepsis[['subject_id','sepsis3']], how='inner', left_on='subject_id', right_on='subject_id')\n",
        "patients = pd.merge(left = patients, right = sepsis[['subject_id','sepsis3']], how='inner', left_on='subject_id', right_on='subject_id')\n",
        "diagnoses_icd = pd.merge(left = diagnoses_icd, right = sepsis[['subject_id','sepsis3']], how='inner', left_on='subject_id', right_on='subject_id')\n",
        "sepsis = pd.merge(left = sepsis, right = icustays[['hadm_id','stay_id']], how='left', left_on='stay_id', right_on='stay_id')"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "sepsis"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 835
        },
        "id": "mSOzdSz_aN6S",
        "outputId": "62d68679-bcf6-408f-9402-08aa467f2c0d"
      },
      "execution_count": 71,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "      subject_id   stay_id      antibiotic_time         culture_time  \\\n",
              "0       16266622  32828424  2134-09-20 16:00:00  2134-09-20 15:32:00   \n",
              "1       13830137  31445489  2174-05-06 22:00:00  2174-05-06 00:00:00   \n",
              "2       14367208  39300189  2135-09-14 18:00:00  2135-09-14 00:45:00   \n",
              "3       12035173  34909617  2168-01-25 02:00:00  2168-01-22 04:04:00   \n",
              "4       13729279  39470047  2181-01-14 04:00:00  2181-01-13 17:50:00   \n",
              "...          ...       ...                  ...                  ...   \n",
              "34673   10827578  34763877  2130-05-09 08:00:00  2130-05-09 11:55:00   \n",
              "34674   11713814  37767813  2159-05-31 08:00:00  2159-05-31 06:25:00   \n",
              "34675   17715294  37357219  2131-07-25 13:00:00  2131-07-25 12:50:00   \n",
              "34676   11180969  35240504  2139-10-30 08:00:00  2139-10-29 21:52:00   \n",
              "34677   16794039  30606871  2112-01-11 00:00:00  2112-01-10 12:29:00   \n",
              "\n",
              "      suspected_infection_time            sofa_time sofa_score respiration  \\\n",
              "0          2134-09-20 15:32:00  2134-09-20 16:00:00          2           2   \n",
              "1          2174-05-06 00:00:00  2174-05-06 21:00:00          2           0   \n",
              "2          2135-09-14 00:45:00  2135-09-14 04:00:00          2           0   \n",
              "3          2168-01-22 04:04:00  2168-01-22 01:00:00          2           0   \n",
              "4          2181-01-13 17:50:00  2181-01-14 05:00:00          2           0   \n",
              "...                        ...                  ...        ...         ...   \n",
              "34673      2130-05-09 08:00:00  2130-05-07 08:00:00         17           3   \n",
              "34674      2159-05-31 06:25:00  2159-05-29 07:00:00         17           2   \n",
              "34675      2131-07-25 12:50:00  2131-07-23 13:00:00         17           2   \n",
              "34676      2139-10-29 21:52:00  2139-10-27 22:00:00         20           4   \n",
              "34677      2112-01-10 12:29:00  2112-01-08 13:00:00         20           4   \n",
              "\n",
              "      coagulation liver cardiovascular cns renal sepsis3   hadm_id  \n",
              "0               0     0              0   0     0    true  23933519  \n",
              "1               0     0              0   0     2    true  26514599  \n",
              "2               0     1              0   0     1    true  21036742  \n",
              "3               0     0              0   2     0    true  29109331  \n",
              "4               0     0              0   2     0    true  24054987  \n",
              "...           ...   ...            ...  ..   ...     ...       ...  \n",
              "34673           3     3              4   0     4    true  20980039  \n",
              "34674           2     2              4   4     3    true  28888207  \n",
              "34675           4     4              4   0     3    true  28399871  \n",
              "34676           2     2              4   4     4    true  26942752  \n",
              "34677           0     4              4   4     4    true  22579763  \n",
              "\n",
              "[34678 rows x 15 columns]"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-5c3f3b24-9a51-4998-85ca-f45c378bd53d\">\n",
              "    <div class=\"colab-df-container\">\n",
              "      <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>subject_id</th>\n",
              "      <th>stay_id</th>\n",
              "      <th>antibiotic_time</th>\n",
              "      <th>culture_time</th>\n",
              "      <th>suspected_infection_time</th>\n",
              "      <th>sofa_time</th>\n",
              "      <th>sofa_score</th>\n",
              "      <th>respiration</th>\n",
              "      <th>coagulation</th>\n",
              "      <th>liver</th>\n",
              "      <th>cardiovascular</th>\n",
              "      <th>cns</th>\n",
              "      <th>renal</th>\n",
              "      <th>sepsis3</th>\n",
              "      <th>hadm_id</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>16266622</td>\n",
              "      <td>32828424</td>\n",
              "      <td>2134-09-20 16:00:00</td>\n",
              "      <td>2134-09-20 15:32:00</td>\n",
              "      <td>2134-09-20 15:32:00</td>\n",
              "      <td>2134-09-20 16:00:00</td>\n",
              "      <td>2</td>\n",
              "      <td>2</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>true</td>\n",
              "      <td>23933519</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>13830137</td>\n",
              "      <td>31445489</td>\n",
              "      <td>2174-05-06 22:00:00</td>\n",
              "      <td>2174-05-06 00:00:00</td>\n",
              "      <td>2174-05-06 00:00:00</td>\n",
              "      <td>2174-05-06 21:00:00</td>\n",
              "      <td>2</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>2</td>\n",
              "      <td>true</td>\n",
              "      <td>26514599</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>14367208</td>\n",
              "      <td>39300189</td>\n",
              "      <td>2135-09-14 18:00:00</td>\n",
              "      <td>2135-09-14 00:45:00</td>\n",
              "      <td>2135-09-14 00:45:00</td>\n",
              "      <td>2135-09-14 04:00:00</td>\n",
              "      <td>2</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>1</td>\n",
              "      <td>true</td>\n",
              "      <td>21036742</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>12035173</td>\n",
              "      <td>34909617</td>\n",
              "      <td>2168-01-25 02:00:00</td>\n",
              "      <td>2168-01-22 04:04:00</td>\n",
              "      <td>2168-01-22 04:04:00</td>\n",
              "      <td>2168-01-22 01:00:00</td>\n",
              "      <td>2</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>2</td>\n",
              "      <td>0</td>\n",
              "      <td>true</td>\n",
              "      <td>29109331</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>13729279</td>\n",
              "      <td>39470047</td>\n",
              "      <td>2181-01-14 04:00:00</td>\n",
              "      <td>2181-01-13 17:50:00</td>\n",
              "      <td>2181-01-13 17:50:00</td>\n",
              "      <td>2181-01-14 05:00:00</td>\n",
              "      <td>2</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>0</td>\n",
              "      <td>2</td>\n",
              "      <td>0</td>\n",
              "      <td>true</td>\n",
              "      <td>24054987</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>34673</th>\n",
              "      <td>10827578</td>\n",
              "      <td>34763877</td>\n",
              "      <td>2130-05-09 08:00:00</td>\n",
              "      <td>2130-05-09 11:55:00</td>\n",
              "      <td>2130-05-09 08:00:00</td>\n",
              "      <td>2130-05-07 08:00:00</td>\n",
              "      <td>17</td>\n",
              "      <td>3</td>\n",
              "      <td>3</td>\n",
              "      <td>3</td>\n",
              "      <td>4</td>\n",
              "      <td>0</td>\n",
              "      <td>4</td>\n",
              "      <td>true</td>\n",
              "      <td>20980039</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>34674</th>\n",
              "      <td>11713814</td>\n",
              "      <td>37767813</td>\n",
              "      <td>2159-05-31 08:00:00</td>\n",
              "      <td>2159-05-31 06:25:00</td>\n",
              "      <td>2159-05-31 06:25:00</td>\n",
              "      <td>2159-05-29 07:00:00</td>\n",
              "      <td>17</td>\n",
              "      <td>2</td>\n",
              "      <td>2</td>\n",
              "      <td>2</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>3</td>\n",
              "      <td>true</td>\n",
              "      <td>28888207</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>34675</th>\n",
              "      <td>17715294</td>\n",
              "      <td>37357219</td>\n",
              "      <td>2131-07-25 13:00:00</td>\n",
              "      <td>2131-07-25 12:50:00</td>\n",
              "      <td>2131-07-25 12:50:00</td>\n",
              "      <td>2131-07-23 13:00:00</td>\n",
              "      <td>17</td>\n",
              "      <td>2</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>0</td>\n",
              "      <td>3</td>\n",
              "      <td>true</td>\n",
              "      <td>28399871</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>34676</th>\n",
              "      <td>11180969</td>\n",
              "      <td>35240504</td>\n",
              "      <td>2139-10-30 08:00:00</td>\n",
              "      <td>2139-10-29 21:52:00</td>\n",
              "      <td>2139-10-29 21:52:00</td>\n",
              "      <td>2139-10-27 22:00:00</td>\n",
              "      <td>20</td>\n",
              "      <td>4</td>\n",
              "      <td>2</td>\n",
              "      <td>2</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>true</td>\n",
              "      <td>26942752</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>34677</th>\n",
              "      <td>16794039</td>\n",
              "      <td>30606871</td>\n",
              "      <td>2112-01-11 00:00:00</td>\n",
              "      <td>2112-01-10 12:29:00</td>\n",
              "      <td>2112-01-10 12:29:00</td>\n",
              "      <td>2112-01-08 13:00:00</td>\n",
              "      <td>20</td>\n",
              "      <td>4</td>\n",
              "      <td>0</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>4</td>\n",
              "      <td>true</td>\n",
              "      <td>22579763</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>34678 rows × 15 columns</p>\n",
              "</div>\n",
              "      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-5c3f3b24-9a51-4998-85ca-f45c378bd53d')\"\n",
              "              title=\"Convert this dataframe to an interactive table.\"\n",
              "              style=\"display:none;\">\n",
              "        \n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "       width=\"24px\">\n",
              "    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n",
              "    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n",
              "  </svg>\n",
              "      </button>\n",
              "      \n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      flex-wrap:wrap;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "      <script>\n",
              "        const buttonEl =\n",
              "          document.querySelector('#df-5c3f3b24-9a51-4998-85ca-f45c378bd53d button.colab-df-convert');\n",
              "        buttonEl.style.display =\n",
              "          google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "        async function convertToInteractive(key) {\n",
              "          const element = document.querySelector('#df-5c3f3b24-9a51-4998-85ca-f45c378bd53d');\n",
              "          const dataTable =\n",
              "            await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                     [key], {});\n",
              "          if (!dataTable) return;\n",
              "\n",
              "          const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "            '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "            + ' to learn more about interactive tables.';\n",
              "          element.innerHTML = '';\n",
              "          dataTable['output_type'] = 'display_data';\n",
              "          await google.colab.output.renderOutput(dataTable, element);\n",
              "          const docLink = document.createElement('div');\n",
              "          docLink.innerHTML = docLinkHtml;\n",
              "          element.appendChild(docLink);\n",
              "        }\n",
              "      </script>\n",
              "    </div>\n",
              "  </div>\n",
              "  "
            ]
          },
          "metadata": {},
          "execution_count": 71
        }
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 72,
      "metadata": {
        "id": "u9muJg8fLS_K"
      },
      "outputs": [],
      "source": [
        "Total_HADMIDS = set(sepsis['hadm_id'])"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "EXTRACT_FEATURE_CODES_VS = dict({\n",
        "    '226754': 'FiO2', '223835': 'FiO2', '227010': 'FiO2', '229280':'FiO2', '229841': 'FiO2',\n",
        "    '220045': 'HR', '220046': 'HR', '226764': 'HR', '227018': 'HR',\n",
        "    '220224': 'PaO2', '227039': 'PaO2', '227516': 'PaO2',\n",
        "    '220050': 'SBP' ,'220179': 'SBP', '224167': 'SBP', '227243': 'SBP', '228152': 'SBP',\n",
        "    '223762': 'Temperature', '226329': 'Temperature', '227632': 'Temperature', '227634': 'Temperature',\n",
        "    })\n",
        "EXTRACT_FEATURE_CODES_LAB = dict({\n",
        "    '220615': 'Creatinine', '229761': 'Creatinine',\n",
        "    '220621': 'Glucose', '225664': 'Glucose', '226537': 'Glucose',\n",
        "    '220228': 'Hb',\n",
        "    '225668': 'Lactacte',\n",
        "    '227457': 'Plt',\n",
        "    '220645': 'Sodium', '226534': 'Sodium',\n",
        "    '220546': 'WBC'}) \n",
        "EXTRACT_FEATURE_CODES_anti = ('225798','225837','225838','225840','225842','225843','225844','225845','225847','225848','225850','225851','225853','225855','225857','225859','225860','225862','225863','225865','225866','225868','225869','225871','225873','225875','225876','225877','225879','225881','225882','225883','225884','225885','225886','225888','225889','225890','225892','225893','225895','225896','225897','225898','225899','225900','225902','225903','225905','227691','228003','229059','229061','229064','229587')\n",
        "\n",
        "EXTRACT_FEATURE_CODES_ANTI_2 = dict({\n",
        "   '225850': 'Cephalosporin', '225851': 'Cephalosporin', '225853': 'Cephalosporin', '225855': 'Cephalosporin', '229587':'Cephalosporin',\n",
        "   '225798': 'Glycopetide',\n",
        "   '225842': 'Penicillin', '225843': 'Penicillin', '225888': 'Penicillin', '225889': 'Penicillin', '225890': 'Penicillin', '225892': 'Penicillin', '225893': 'Penicillin',\n",
        "   '225859': 'Quninolone', '225859': 'Quninolone', '225879': 'Quninolone', '225886': 'Quninolone',\n",
        "   '225845': 'Macrolide', '225865': 'Macrolide', '225866': 'Macrolide'\n",
        "})"
      ],
      "metadata": {
        "id": "Yt6qYxMBwAtn"
      },
      "execution_count": 73,
      "outputs": []
    },
    {
      "cell_type": "code",
      "execution_count": 14,
      "metadata": {
        "id": "3CtnHsoVFrlR"
      },
      "outputs": [],
      "source": [
        "# EXTRACT_FEATURE_CODES_VS = dict({\n",
        "#     '220045': 'PULSE',\n",
        "#     '211': 'PULSE',\n",
        "#     '618': 'RESP',\n",
        "#     '220210': 'RESP',\n",
        "#     '220050': 'SBP',\n",
        "#     '220179': 'SBP',\n",
        "#     '225309': 'SBP',\n",
        "#     '51': 'SBP',\n",
        "#     '455': 'SBP',\n",
        "#     '220051': 'DBP',\n",
        "#     '220180': 'DBP',\n",
        "#     '225310': 'DBP',\n",
        "#     '8368': 'DBP',\n",
        "#     '8441': 'DBP',\n",
        "#     '223762': 'TEMP',\n",
        "#     '676': 'TEMP',\n",
        "#     '677': 'TEMP',\n",
        "#     '223761': 'TEMP_F',\n",
        "#     '678': 'TEMP_F',\n",
        "#     '679': 'TEMP_F',\n",
        "#     '223835' : 'FiO2',\n",
        "#     '189' : 'FiO2',\n",
        "#     '190' : 'FiO2',\n",
        "#     '220052' : 'MAP',\n",
        "#     '1321' : 'MAP',\n",
        "#     '438' : 'MAP'\n",
        "# })\n",
        "\n",
        "# EXTRACT_FEATURE_CODES_LAB = dict({\n",
        "#     '50885': 'BILIRUBIN',\n",
        "#     '220546': 'WBC',\n",
        "#     '51301': 'WBC',\n",
        "#     '51221': 'HEMATOCRIT',\n",
        "#     '225668': 'LACTATE',\n",
        "#     '50813': 'LACTATE',\n",
        "#     '51265': 'PLATELET',\n",
        "#     '50912': 'CREATININE',\n",
        "#     '220615': 'CREATININE',\n",
        "#     '229761': 'CREATININE',\n",
        "#     '50983': 'SODIUM',\n",
        "#     '50820': 'PH',\n",
        "#     '50831': 'PH',\n",
        "#     '50971': 'POTASSIUM'\n",
        "# })\n",
        "\n",
        "FEATURE_NORMAL_RANGE = dict({\n",
        "    'PULSE': (0, 300),\n",
        "    'SBP': (0, 300),\n",
        "    'DBP': (0, 300),\n",
        "    'RESP': (0, 120),\n",
        "    'TEMP': (25, 50),\n",
        "    'BILIRUBIN': (0, 75),\n",
        "    'WBC': (0, 100),\n",
        "    'HEMATOCRIT': (0, 100),\n",
        "    'LACTATE': (0, 22),\n",
        "    'PLATELET': (0, 1000),\n",
        "    'CREATININE': (0, 20),\n",
        "    'SODIUM': (0, 500) \n",
        "})\n",
        "\n",
        "# VITAL_SIGNS = {'PULSE': ['220045', '211'],\n",
        "#                'RESP': ['220210', '618'],\n",
        "#                'SBP': ['220050', '220179', '225309', '51', '455'],\n",
        "#                'DBP': ['220051', '220180', '225310', '8368', '8441'],\n",
        "#                'FiO2': ['223835', '189', '190'],\n",
        "#                'MAP': ['220052', '1321', '438']\n",
        "#                }\n",
        "# LABS = {'BILIRUBIN': ['50885'],\n",
        "#         'LACTATE': ['50813'],\n",
        "#         'pH': ['50820', '50831'],\n",
        "#         'SODIUM': ['50983'],\n",
        "#         'POTASSIUM': ['50971'],\n",
        "#         'CREATININE': ['50912'],\n",
        "#         'WBC': ['51301'],\n",
        "#         'HEMATOCRIT': ['51221']\n",
        "#         }\n",
        "\n",
        "# VITAL_SIGNS = {'PULSE': ['220045', '211'],\n",
        "#                'RESP': ['220210', '618'],\n",
        "#                'SBP': ['228152','229899','220050', '220179', '225309', '51', '455','224167','227243'],\n",
        "#                'DBP': ['220051', '220180', '225310', '8368', '8441'],\n",
        "#                'FiO2': ['229280','229841','226754','227009','227010', '223835', '189', '190'],\n",
        "#                'MAP': ['220052', '1321', '438']\n",
        "#                }\n",
        "\n",
        "# LABS = {'lactate': ['225668'],\n",
        "#         'glucose'\t: ['220621','225664','226537','228388'],\n",
        "#         'creatinine' : ['220615','229761'],\n",
        "#         'WBC'\t: ['220546','51301'],\n",
        "#         'heart rate' : ['220046','220047','226764','220045','227018'],\n",
        "#         'sodium' : ['220645','226534','228389','228390'],\n",
        "#         'temperature' : ['223762','226329','227632','227634'],\n",
        "#         'platelet' : ['227457'],\n",
        "#         'hemoglobin' : ['220228'],\n",
        "#         'pao2' : ['220224','227516','227039'],\n",
        "# }"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 10,
      "metadata": {
        "id": "ZwMLOPrMI-6d"
      },
      "outputs": [],
      "source": [
        "## Define some util functions\n",
        "NOT_CONVERTED = 'NOT_CONVERTED'\n",
        "## 문자로 기록된 날짜 값들을 파이썬의 datetime 데이터로 바꿔주는 함수 작성\n",
        "def str2datetime(s):\n",
        "    def _convert(_s, _dformat):\n",
        "        try:\n",
        "            converted_dt = datetime.strptime(_s, _dformat)\n",
        "        except Exception:\n",
        "            return NOT_CONVERTED\n",
        "\n",
        "        return converted_dt\n",
        "\n",
        "    if isinstance(s, datetime):\n",
        "        return s\n",
        "\n",
        "    dformats = [\n",
        "        '%Y-%m-%d %p %I:%M:%S',\n",
        "        '%Y-%m-%d %H:%M:%S.%f',\n",
        "        '%Y-%m-%d %H:%M:%S',\n",
        "        '%Y-%m-%d %H:%M',\n",
        "        '%y-%m-%d %H:%M',\n",
        "        '%Y-%m-%d',\n",
        "        '%Y%m%d',\n",
        "        '%Y-%m-%d +%H:%M',\n",
        "        '%Y%m%d %H%M',\n",
        "        '%Y%m%d%H%M%S',\n",
        "        ]\n",
        "\n",
        "    s = s.split('.')[0]\n",
        "    s = s.replace('오전', 'am').replace('오후', 'pm')\n",
        "    for dformat in dformats:\n",
        "        dt = _convert(s, dformat)\n",
        "        if dt != NOT_CONVERTED:\n",
        "            return dt\n",
        "\n",
        "    return NOT_CONVERTED"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 11,
      "metadata": {
        "id": "l4eGWeCbGRQR"
      },
      "outputs": [],
      "source": [
        "def extract_features():\n",
        "  features_hadm_id = dict()\n",
        "  chartevents = csv.DictReader(open('/content/gdrive/MyDrive/mimiciv_unzip/icu/chartevents.csv', 'r', encoding='utf-8'))\n",
        "  labevents = csv.DictReader(open('/content/gdrive/MyDrive/mimiciv_unzip/hosp/labevents.csv', 'r', encoding='utf-8'))\n",
        "  for row in tqdm(chartevents, desc='Extract_VS', total=7241351, ncols=700):\n",
        "    if row['hadm_id'] in Total_HADMIDS:\n",
        "      hadm_id = row['hadm_id']\n",
        "      features_hadm_id[hadm_id] = features_hadm_id.get(hadm_id, dict()) #get 없으면 2번째로 return\n",
        "\n",
        "      item_id = row['itemid']\n",
        "      if item_id not in EXTRACT_FEATURE_CODES_VS:\n",
        "        continue\n",
        "      item_name = EXTRACT_FEATURE_CODES_VS[item_id]\n",
        "      try:\n",
        "        #정규식을 이용해서 기록된 값에서 숫자 형태의 값만 가져오기\n",
        "        #소수점 둘째자리 까지만 값을 사용하기\n",
        "        item_value = round(float(re.findall(\"\\d+\\.\\d+|\\d+\", row['valuenum'])[0]), 2)\n",
        "      except Exception:\n",
        "        continue\n",
        "      \n",
        "      item_time = str2datetime(row['charttime'])\n",
        "      if item_time == NOT_CONVERTED:\n",
        "        continue\n",
        "      #GCP total score도 시간대에 총합으로 사용함\n",
        "      if item_name == 'TEMP_F':\n",
        "        item_value = round((item_value - 32.0) / 1.8, 2)\n",
        "        item_name = 'TEMP'\n",
        "\n",
        "    #   if item_value > FEATURE_NORMAL_RANGE[item_name][1]:\n",
        "    #     item_value = FEATURE_NORMAL_RANGE[item_name][1]\n",
        "    #   elif item_value < FEATURE_NORMAL_RANGE[item_name][0]:\n",
        "    #     item_value = FEATURE_NORMAL_RANGE[item_name][0]\n",
        "\n",
        "      features_hadm_id[hadm_id][item_name] = features_hadm_id[hadm_id].get(item_name, list()) #같은 환자의 여러개 feature가 있으면 list로 반환하도록\n",
        "      features_hadm_id[hadm_id][item_name].append((item_time, item_value))\n",
        "      \n",
        "    else:\n",
        "      continue\n",
        "\n",
        "  for row in tqdm(labevents, desc='Extract_LAB', total=615328, ncols=700):\n",
        "    if row['hadm_id'] in Total_HADMIDS:\n",
        "      hadm_id = row['hadm_id']\n",
        "      features_hadm_id[hadm_id] = features_hadm_id.get(hadm_id, dict())\n",
        "\n",
        "      item_id = row['itemid']\n",
        "      if item_id not in EXTRACT_FEATURE_CODES_LAB:\n",
        "        continue\n",
        "      item_name = EXTRACT_FEATURE_CODES_LAB[item_id]\n",
        "      try:\n",
        "        #정규식을 이용해서 기록된 값에서 숫자 형태의 값만 가져오기\n",
        "        #소수점 둘째자리 까지만 값을 사용하기\n",
        "        item_value = round(float(re.findall(\"\\d+\\.\\d+|\\d+\", row['valuenum'])[0]), 2)\n",
        "      except Exception:\n",
        "        continue\n",
        "      \n",
        "      item_time = str2datetime(row['charttime'])\n",
        "      if item_time == NOT_CONVERTED:\n",
        "        continue\n",
        "\n",
        "      #if item_value > FEATURE_NORMAL_RANGE[item_name][1]:\n",
        "      #  item_value = FEATURE_NORMAL_RANGE[item_name][1]\n",
        "      #elif item_value < FEATURE_NORMAL_RANGE[item_name][0]:\n",
        "      #  item_value = FEATURE_NORMAL_RANGE[item_name][0]\n",
        "        \n",
        "      features_hadm_id[hadm_id][item_name] = features_hadm_id[hadm_id].get(item_name, list())\n",
        "      features_hadm_id[hadm_id][item_name].append((item_time, item_value))\n",
        "      \n",
        "    else:\n",
        "      continue\n",
        "      \n",
        "  return features_hadm_id"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": 15,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 81,
          "referenced_widgets": [
            "5472ca8f6e3a403f91739d19b8faea98",
            "976b2e903f2049bf927d87eb31eafa31",
            "68643b54cbc74e93a8348379aa6652dc",
            "79e802f0d027404cb608633f20c01207",
            "6c20ab6bf17f4ea0b8f7b3a60352fe59",
            "3b0066087ebf403fb7553faed14e3cad",
            "0f552905a4f74370959d443c767997a1",
            "fc4da944755b4bec9cc17810bd45cdcc",
            "648fd305dd104142b305433c56798fae",
            "2403faf1c0cc4ef8bcfcd64e99b255fa",
            "eabd8a510ecc48f9a494452885a3de3d",
            "f326faeebfc84cf6baeb0f1e35cb29af",
            "2036c48bed07415a9f6d733d1899cf04",
            "3f62d85a1c2d4031bf9f1b3a80f454e7",
            "8cfee1ca14a84ae88d5d47d6c2a4874f",
            "105ad6d8432b483d8ee44824a89e3603",
            "24c16358d2894ab191fe8d976fec2b8f",
            "8fa7f9fec3514301ab191875dcd17ae0",
            "5e64c2da6e464fa6871b7e2594c55b27",
            "bdfd3e40c1f94232bba5b112cd5b94ce",
            "f59ec96981994f8288e939c4bf411f99",
            "6882ecbd9730474dad27c0a7888cf7df"
          ]
        },
        "id": "8IuFnUqELI10",
        "outputId": "8c7c7316-686c-4b2a-8f7f-ecdc6656d586"
      },
      "outputs": [
        {
          "data": {
            "application/vnd.jupyter.widget-view+json": {
              "model_id": "5472ca8f6e3a403f91739d19b8faea98",
              "version_major": 2,
              "version_minor": 0
            },
            "text/plain": [
              "Extract_VS:   0%|                                                                                             …"
            ]
          },
          "metadata": {},
          "output_type": "display_data"
        },
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Extract_LAB:   0%|                                                                                            …"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "f326faeebfc84cf6baeb0f1e35cb29af"
            }
          },
          "metadata": {}
        }
      ],
      "source": [
        "extracted_features = extract_features()"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# import pickle\n",
        "# import gzip\n",
        "# with gzip.open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_extracted_features_raw.pickle', 'wb') as f:\n",
        "#     pickle.dump(extracted_features, f)"
      ],
      "metadata": {
        "id": "ENvqmlMyO_pk"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# from itertools import islice\n",
        "# dict(list(extracted_features.items())[0:2])"
      ],
      "metadata": {
        "id": "V1eteo_G_Fkj"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# 사망 레이블 하기\n",
        "death_labels = dict()\n",
        "admissions = csv.DictReader(open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_admissions.csv', 'r', encoding='utf-8'))\n",
        "for row in tqdm(admissions, desc='Get Death Label'):\n",
        "  hadm_id = row['hadm_id']\n",
        "  death_yn = int(row['hospital_expire_flag'])\n",
        "  if death_yn:\n",
        "    event_time = str2datetime(row['deathtime'])\n",
        "  else:\n",
        "    event_time = str2datetime(row['dischtime'])\n",
        "  death_labels[hadm_id] = dict({'DEATH': death_yn, 'EVENT_TIME': event_time})"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 49,
          "referenced_widgets": [
            "99a409963f2e4bd28f8b6cb935971a90",
            "aab2360df1ef43eb85eba7c663aa37eb",
            "30be9461262e4a8b832c5c7d92891256",
            "1627cc744fde4d9fbe3a33efaa8834ec",
            "6b20b1dc24ee4fa0944cb80e8727ed5a",
            "c34cebd5d4e34e318dcbfdae81d7e86d",
            "20856359ee754f51a5a03d59dc078815",
            "3e6d9b592e5e41a8aaf7620ee4fe84c5",
            "5fecb08f5f5a43018377ca399bc57f65",
            "efa07aac08a14f5a89ba4a2e2b8b3071",
            "6f4c931c45d54eccb26f5a64ccbd321c"
          ]
        },
        "id": "GUHXlnCqNTjo",
        "outputId": "6d24fec2-ac72-413e-82f3-b896814614c9"
      },
      "execution_count": 12,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Get Death Label: 0it [00:00, ?it/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "99a409963f2e4bd28f8b6cb935971a90"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# 재실 기간 레이블 하기\n",
        "los_labels = dict()\n",
        "admissions = csv.DictReader(open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_admissions.csv', 'r', encoding='utf-8'))\n",
        "for row in tqdm(admissions, desc='Get loss Label'):\n",
        "  hadm_id = row['hadm_id']\n",
        "  admin_time = str2datetime(row['admittime'])\n",
        "  disc_time = str2datetime(row['dischtime'])\n",
        "  los = (disc_time - admin_time).total_seconds() / (60*60)\n",
        "  los_labels[hadm_id] = int(los)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 49,
          "referenced_widgets": [
            "3a814315b42c43f4a59cd571186189f6",
            "906c72d94c1d495bbad9ed08917f9b83",
            "15f42cc8f32c4dd198e255c8e6b78217",
            "3bcc410f5b21483fa345c4cf1f2f0af4",
            "f6437692ca0d4c049423f80706fc24a6",
            "c16ca63c7851488eb13d26f99058b86c",
            "94a2168440b14a978d5dc17a309c06c8",
            "4fe63e54e192470b91014fb139d84e63",
            "77e4e875434a4d54adedc02ea6630ce4",
            "f033a915a1fe40b2afbcf593694c3e30",
            "fbc98104317d4d40947e6c4cb7561109"
          ]
        },
        "id": "PtXfAzo0i7Hq",
        "outputId": "582843ac-68a5-40e0-d628-2a6aaa908fd7"
      },
      "execution_count": 13,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Get loss Label: 0it [00:00, ?it/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "3a814315b42c43f4a59cd571186189f6"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "## 환자의 입실 퇴실 시간 정보 가져오기\n",
        "Extracted_HADMIDS = list(extracted_features.keys())\n",
        "admin_disc_time = dict()\n",
        "\n",
        "admissions = csv.DictReader(open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_admissions.csv', 'r', encoding='utf-8'))\n",
        "for row in admissions:\n",
        "  hadm_id = row['hadm_id']\n",
        "  admin_time = str2datetime(row['admittime'])\n",
        "  disc_time = str2datetime(row['dischtime'])\n",
        "\n",
        "  if (admin_time != NOT_CONVERTED) and (disc_time != NOT_CONVERTED):\n",
        "    admin_disc_time[hadm_id] = dict({'ADMIN': admin_time, 'DISC': disc_time})"
      ],
      "metadata": {
        "id": "Zm7qGtNPOH3p"
      },
      "execution_count": 16,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "## 입실 시점으로부터 1시간 간격으로 데이터를 읽어오는 함수 정의\n",
        "def quantize_features(feature_list, admin_time, disc_time, missing_feature=False):\n",
        "  total_hours = int((disc_time - admin_time).total_seconds() / (60*60))\n",
        "  #입실 이후 최대 24시간 동안의 데이터만 가져오기\n",
        "  if total_hours > 24:\n",
        "    total_hours = 24\n",
        "  \n",
        "  quantized_features = list()\n",
        "  if missing_feature:\n",
        "    for i in range(total_hours):\n",
        "      quantized_time = admin_time + timedelta(hours=i)\n",
        "      quantized_value = np.nan #지금은 dic 형태라, pandas 쓰면 NAN이랑 동일, carry forward 도 가능\n",
        "      quantized_features.append((quantized_time, quantized_value))\n",
        "  else:\n",
        "    feature_list = list(sorted(feature_list, key=lambda x: x[0]))\n",
        "    for i in range(total_hours):\n",
        "      quantized_time = admin_time + timedelta(hours=i)\n",
        "      quantized_time_start = quantized_time - timedelta(hours=1)\n",
        "      quantized_values = list(filter(lambda x: quantized_time_start < x[0] <= quantized_time, feature_list)) #filter함수 True만 필터\n",
        "      if len(quantized_values) > 0:\n",
        "        quantized_value = quantized_values[-1][1]\n",
        "      else:\n",
        "        quantized_value = np.nan \n",
        "\n",
        "      quantized_features.append((quantized_time, quantized_value))\n",
        "  \n",
        "  return quantized_features"
      ],
      "metadata": {
        "id": "Auy7XQu0N5cN"
      },
      "execution_count": 17,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "## 전체 데이터에 대해서 데이터 변환하기\n",
        "quantized_features = dict()\n",
        "for hadm_id in tqdm(Extracted_HADMIDS, desc='Doing Quantization'):\n",
        "    feature_keys = ['PULSE', 'SBP', 'DBP', 'RESP', 'TEMP','FiO2','MAP', 'BILIRUBIN', 'WBC', 'HEMATOCRIT','PH' ,'LACTATE', 'PLATELET', 'CREATININE', 'SODIUM','POTASSIUM']\n",
        "    admin_time = admin_disc_time[hadm_id]['ADMIN']\n",
        "    disc_time = admin_disc_time[hadm_id]['DISC']\n",
        "    if (disc_time - admin_time).total_seconds() / (60*60) < 1.0:\n",
        "      continue\n",
        "    quantized_features[hadm_id] = dict()\n",
        "    for kk in feature_keys:\n",
        "      if kk in extracted_features[hadm_id]:\n",
        "        feature_list = extracted_features[hadm_id][kk]\n",
        "        quantized_features[hadm_id][kk] = quantize_features(feature_list, admin_time, disc_time)\n",
        "      else:\n",
        "        quantized_features[hadm_id][kk] = quantize_features(None, admin_time, disc_time, missing_feature=True)\n"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 49,
          "referenced_widgets": [
            "9debacc05af94edbaee6d215adce9028",
            "c43951b1593c4b69af720286ceacf623",
            "47ee97ea6f424167a79ae31e3b084e2b",
            "717dd6e408294aa995460682eb8bda6e",
            "fb9ae9afd01942c48ea79f999629943e",
            "daed62cfa52945a5b8ba2d4729f7d467",
            "867f5717f32442eca389a33478203f7b",
            "8560981b4fe544018079359126904fb1",
            "2747112928424905afdc9f1af291e344",
            "29067893497e418b848a0055ac5049d3",
            "54c37704080346ffbceaaa08cbab6e7d"
          ]
        },
        "id": "SxMcge9WRy-m",
        "outputId": "d537240a-3cb2-447f-fa11-74507964e28e"
      },
      "execution_count": 18,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Doing Quantization:   0%|          | 0/32812 [00:00<?, ?it/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "9debacc05af94edbaee6d215adce9028"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "NORMAL_VALUE = dict({\n",
        "    'PULSE': 80.0,\n",
        "    'SBP': 120.0,\n",
        "    'DBP': 80.0,\n",
        "    'RESP': 14.0,\n",
        "    'TEMP': 36.5,\n",
        "    'BILIRUBIN': 0.6,\n",
        "    'WBC': 7.0,\n",
        "    'HEMATOCRIT': 45.0,\n",
        "    'LACTATE': 0.7,\n",
        "    'PLATELET': 300.,\n",
        "    'CREATININE': 0.8,\n",
        "    'SODIUM': 140.,\n",
        "    'FiO2' : 400,\n",
        "    'PH' : 7.35,\n",
        "    'POTASSIUM' : 4.25,\n",
        "    'MAP' : 85\n",
        "})"
      ],
      "metadata": {
        "id": "FP4ouD6ISoe5"
      },
      "execution_count": 19,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "def carry_forward(feature_name, feature_list):\n",
        "  imputed_features = list()\n",
        "  for i, (feature_t, feature_v) in enumerate(feature_list):\n",
        "    if np.isnan(feature_v):\n",
        "      last_observations = list(filter(lambda x: (x[0] <= feature_t) and (not np.isnan(x[1])), feature_list))\n",
        "      if len(last_observations) > 0:\n",
        "        imputed_features.append((feature_t, last_observations[-1][1]))\n",
        "      else:\n",
        "        imputed_features.append((feature_t, NORMAL_VALUE[feature_name]))\n",
        "    else:\n",
        "      imputed_features.append((feature_t, feature_v))\n",
        "\n",
        "  return imputed_features"
      ],
      "metadata": {
        "id": "th7CkR4pSJE8"
      },
      "execution_count": 20,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "Quantized_HADMIDS = list(quantized_features.keys())\n",
        "carry_forwarded_features = dict()\n",
        "feature_keys = ['PULSE', 'SBP', 'DBP', 'RESP', 'TEMP','FiO2','MAP', 'BILIRUBIN', 'WBC', 'HEMATOCRIT','PH' ,'LACTATE', 'PLATELET', 'CREATININE', 'SODIUM','POTASSIUM']\n",
        "for hadm_id in tqdm(Quantized_HADMIDS, desc='Carry-Forward'):\n",
        "  carry_forwarded_features[hadm_id] = dict()\n",
        "\n",
        "  for kk in feature_keys:\n",
        "    feature_list = quantized_features[hadm_id][kk]\n",
        "    carry_forwarded_features[hadm_id][kk] = carry_forward(kk, feature_list)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 49,
          "referenced_widgets": [
            "eae23d3ecca048dfbe0972630280425e",
            "b1338e8b13d340bfbd5923558b12d9af",
            "906bb79f38314401b11ae2932482d53a",
            "832c89ff71714c0b96187a45020041c4",
            "387570612c284612888c0e020f28cbf6",
            "7a8175aa904d40b8973239cca8b9d530",
            "79d42ecf68984391b49b58c4238085de",
            "57d69c105ef44ecdaed386ac4063697d",
            "837fbb0216e74cf889e32af4a77d9dc9",
            "35aa6271461d4529b669e4327b6433ef",
            "01ca384bcbb145f1885c96da3eb83b82"
          ]
        },
        "id": "uhmbOBT_Shgz",
        "outputId": "bc02628b-f9aa-4b35-920a-b4b8857e3362"
      },
      "execution_count": 21,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Carry-Forward:   0%|          | 0/32785 [00:00<?, ?it/s]"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "eae23d3ecca048dfbe0972630280425e"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#항목별 정상값의 범위\n",
        "#수정 필요\n",
        "FEATURE_NORMAL_RANGE = dict({\n",
        "    'PULSE': (0, 300),\n",
        "    'SBP': (0, 300),\n",
        "    'DBP': (0, 300),\n",
        "    'RESP': (0, 120),\n",
        "    'TEMP': (25, 50),\n",
        "    'BILIRUBIN': (0, 75),\n",
        "    'WBC': (0, 100),\n",
        "    'HEMATOCRIT': (0, 100),\n",
        "    'LACTATE': (0, 20),\n",
        "    'PLATELET': (0, 1000),\n",
        "    'CREATININE': (0, 20),\n",
        "    'SODIUM': (0, 500),\n",
        "    'FiO2' : (300, 500),\n",
        "    'PH' : (7.35, 7.45),\n",
        "    'POTASSIUM' : (3.7, 5.2),\n",
        "    'MAP' : (70, 100)\n",
        "})"
      ],
      "metadata": {
        "id": "IKibj9tsl_JV"
      },
      "execution_count": 22,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "#Min-Max Normalization\n",
        "min_max_normalized = dict()\n",
        "for hadm_id in tqdm(Quantized_HADMIDS, desc='Min-Max Normalization', ncols=700):\n",
        "  min_max_normalized[hadm_id] = dict()\n",
        "  feature_keys = ['PULSE', 'SBP', 'DBP', 'RESP', 'TEMP','FiO2','MAP', 'BILIRUBIN', 'WBC', 'HEMATOCRIT','PH' ,'LACTATE', 'PLATELET', 'CREATININE', 'SODIUM','POTASSIUM']\n",
        "  for kk in feature_keys:\n",
        "    features = np.array(carry_forwarded_features[hadm_id][kk])\n",
        "\n",
        "    min = FEATURE_NORMAL_RANGE[kk][0]\n",
        "    max = FEATURE_NORMAL_RANGE[kk][1]\n",
        "\n",
        "    features[:, 1] = (features[:, 1] - min) / (max - min)\n",
        "    min_max_normalized[hadm_id][kk] = features"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 49,
          "referenced_widgets": [
            "fffe02e0eacb404580aa1de153e0a157",
            "7bc518e6d26146c98226a28a2a7363ef",
            "dba7241eee804b56b52f010b75401fc1",
            "276fc4ae40f64a1696b2843e5e10c116",
            "c765eb6b4423459c8eac49d006c50623",
            "a24e7ccb9b1c4659ab6453c5e854631c",
            "e41cedbe0cd443efb48c5d3c2e107ce5",
            "e44e6718c2ee4616b6aa380e03716466",
            "98004a3bdd344ef09983c8ffe04298af",
            "ef586eeca984410f8778eac0f736b40c",
            "e266f2c4a2fc4aa59686397701a72695"
          ]
        },
        "id": "uOusHtWwTGAV",
        "outputId": "00539bf6-34e6-403c-bae8-af97ec6c6236"
      },
      "execution_count": 23,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Min-Max Normalization:   0%|                                                                                  …"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "fffe02e0eacb404580aa1de153e0a157"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "total_death_included_hadm_ids = list(death_labels.keys())\n",
        "total_los_included_hadm_ids = list(los_labels.keys())\n",
        "feature_keys = ['PULSE', 'SBP', 'DBP', 'RESP', 'TEMP','FiO2','MAP', 'BILIRUBIN', 'WBC', 'HEMATOCRIT','PH' ,'LACTATE', 'PLATELET', 'CREATININE', 'SODIUM','POTASSIUM']"
      ],
      "metadata": {
        "id": "Tg6sx0ytTLzE"
      },
      "execution_count": 24,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "## 사망 예측용 데이터\n",
        "death_dataset = dict({\n",
        "    'HADM_IDS': list(),\n",
        "    'LABEL': list(),\n",
        "    'DATA': list()\n",
        "})\n",
        "\n",
        "for hadm_id in tqdm(total_death_included_hadm_ids, desc='Make Death Dataset', ncols=700):\n",
        "  if hadm_id not in min_max_normalized:\n",
        "    continue\n",
        "  if len(min_max_normalized[hadm_id][feature_keys[0]]) >= 24:\n",
        "    time_len = 24\n",
        "  else:\n",
        "    time_len = len(min_max_normalized[hadm_id][feature_keys[0]])\n",
        "  \n",
        "  try:\n",
        "    data = np.array(min_max_normalized[hadm_id][feature_keys[0]])[:time_len, 1].reshape(time_len, 1)\n",
        "  except:\n",
        "    continue\n",
        "\n",
        "  for kk in feature_keys[1:]:\n",
        "    data = np.concatenate((data, np.array(min_max_normalized[hadm_id][kk])[:time_len, 1].reshape(time_len, 1)), axis=1)\n",
        "  \n",
        "  death_dataset['HADM_IDS'].append(hadm_id)\n",
        "  death_dataset['LABEL'].append(death_labels[hadm_id]['DEATH'])\n",
        "  death_dataset['DATA'].append(data)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 49,
          "referenced_widgets": [
            "716537a7fcb2471e9857bba28c484cdd",
            "a5a2f326cf9946d89bddeaf0ccb714b1",
            "62681f4b05e74186b91489e682150fe4",
            "e870357a5c7a41db89ffa596e99f6b3d",
            "fea6c742f7dd46ea8836080c90079aaa",
            "5ce939731df9495bb4c430157f35a2d5",
            "650a30f0863746318784885af11cef7b",
            "4de03ebccf714493b1d0fcae20bfdc8b",
            "91a9e58cbbbf4b3aae6e4e161713b3b9",
            "b99411094dc5458d850b8aa17649e2e5",
            "6341951516234d30a3a9dd900856db6f"
          ]
        },
        "id": "GLGBtCnTk0XN",
        "outputId": "5fd32927-a238-4fc8-fdf4-380d359aaf7b"
      },
      "execution_count": 25,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Make Death Dataset:   0%|                                                                                     …"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "716537a7fcb2471e9857bba28c484cdd"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "## 재실기간 예측용 데이터\n",
        "los_dataset = dict({\n",
        "    'HADM_IDS': list(),\n",
        "    'LABEL': list(),\n",
        "    'DATA': list()\n",
        "})\n",
        "\n",
        "for hadm_id in tqdm(total_los_included_hadm_ids, desc='Make LOS Dataset', ncols=700):\n",
        "  time_len = 12\n",
        "  \n",
        "  try:\n",
        "    data = np.array(min_max_normalized[hadm_id][feature_keys[0]])[:time_len, 1].reshape(time_len, 1)\n",
        "  except:\n",
        "    continue\n",
        "\n",
        "  for kk in feature_keys[1:]:\n",
        "    data = np.concatenate((data, np.array(min_max_normalized[hadm_id][kk])[:time_len, 1].reshape(time_len, 1)), axis=1)\n",
        "  \n",
        "  los_dataset['HADM_IDS'].append(hadm_id)\n",
        "  los_dataset['LABEL'].append(los_labels[hadm_id])\n",
        "  los_dataset['DATA'].append(data)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 49,
          "referenced_widgets": [
            "120ca2b8d38f411496c6bb72b653d090",
            "7af63be1ccd14f708b4b77f2fbb38b26",
            "c72a6d8b5cbf4c70a25bf1330c872d2e",
            "9041ccad9bb24822be512ba6e582ef09",
            "f0483655f18943e6a1631b62395b704a",
            "af624fee48934f2a9c4001b27ad5faa7",
            "2cfabc724ae64c2ab78a2383f47ec6a9",
            "6c507a638fb84705b680d22db7b20d27",
            "f7bb27ca58704a0c93f03e31ae1dcddd",
            "07194774c7d541eea5e749d98dce11c3",
            "89ebf360b3f74a17829e4cf762d4dfbd"
          ]
        },
        "id": "jQUuy4y9qEjl",
        "outputId": "32025bc8-db78-4669-f01e-519fc456bf53"
      },
      "execution_count": 26,
      "outputs": [
        {
          "output_type": "display_data",
          "data": {
            "text/plain": [
              "Make LOS Dataset:   0%|                                                                                       …"
            ],
            "application/vnd.jupyter.widget-view+json": {
              "version_major": 2,
              "version_minor": 0,
              "model_id": "120ca2b8d38f411496c6bb72b653d090"
            }
          },
          "metadata": {}
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "# # 항생제 레이블링하기\n",
        "# EXTRACT_FEATURE_CODES_anti2\n",
        "# anti_labels = dict()\n",
        "# chartevents = csv.DictReader(open('/content/gdrive/MyDrive/mimiciv_unzip/icu/chartevents.csv', 'r', encoding='utf-8'))\n",
        "# for row in tqdm(chartevents, desc='Get anti Label'):\n",
        "#   hadm_id = row['hadm_id']\n",
        "#   if row['itemid'] in EXTRACT_FEATURE_CODES_anti:\n",
        "#     anti_label_group = row['itemid'][0]\n",
        "#     anti_label_time = str2datetime(row['charttime'][0])\n",
        "#   else:\n",
        "#     anti_label_group = 'none'\n",
        "#     anti_label_time = 'none'\n",
        "#   anti_labels[hadm_id] = dict({'anti': anti_label_group, 'anti_TIME': anti_label_time})\n",
        "#   # 항생제 레이블링하기\n",
        "# anti_labels = dict()\n",
        "# chartevents = csv.DictReader(open('/content/gdrive/MyDrive/mimiciv_unzip/icu/chartevents.csv', 'r', encoding='utf-8'))\n",
        "# for row in tqdm(chartevents, desc='Get anti Label'):\n",
        "#   hadm_id = row['hadm_id']\n",
        "#   if row['itemid'] in EXTRACT_FEATURE_CODES_anti:\n",
        "#     anti_label_group = row['itemid'][0]\n",
        "#     anti_label_time = str2datetime(row['charttime'][0])\n",
        "#   else:\n",
        "#     anti_label_group = 'none'\n",
        "#     anti_label_time = 'none'\n",
        "#   anti_labels[hadm_id] = dict({'anti': anti_label_group, 'anti_TIME': anti_label_time})\n",
        "#   anti_labels.values()\n",
        "# from itertools import islice\n",
        "# dict(list(anti_labels.items())[0:2])"
      ],
      "metadata": {
        "id": "P7TKXba63Qro"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "# with open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_dataset_death_HADM_IDS.p', 'wb') as f:\n",
        "#     pickle.dump(death_dataset['HADM_IDS'], f)\n",
        "# with open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_dataset_death_LABEL.p', 'wb') as f:\n",
        "#     pickle.dump(death_dataset['LABEL'], f)\n",
        "# with open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_dataset_death_DATA.p', 'wb') as f:\n",
        "#     pickle.dump(death_dataset['DATA'], f)\n",
        "# death_dataset_read = dict({\n",
        "#     'HADM_IDS': list(),\n",
        "#     'LABEL': list(),\n",
        "#     'DATA': list()\n",
        "# })\n",
        "# with open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_dataset_death_HADM_IDS.p', 'rb') as f:\n",
        "#     death_dataset_read['HADM_IDS'] = pickle.load(f)\n",
        "# #load\n",
        "# with open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_dataset_death.pickle', 'rb') as f:\n",
        "#     death_dataset_read = pickle.load(f)\n",
        "\n",
        "# # gzip 압축 파일을 열기\n",
        "# with gzip.open('/content/gdrive/MyDrive/mimiciv_der/sepsis3_dataset_death.pickle.gz', 'rb') as f:\n",
        "#     death_dataset_read = pickle.load(f)"
      ],
      "metadata": {
        "id": "mQxcsHPUzvte"
      },
      "execution_count": null,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "pd.DataFrame(zip(death_dataset['HADM_IDS'],death_dataset['LABEL'] ,death_dataset['DATA'])).head()"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 206
        },
        "id": "KpmQMOuxqqVi",
        "outputId": "9cceb80d-d5e6-4c03-e503-97acb5a5a3b0"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "          0  1                                                  2\n",
              "0  26184834  1  [[0.26666666666666666, 0.4, 0.2666666666666666...\n",
              "1  23581541  0  [[0.26666666666666666, 0.4, 0.2666666666666666...\n",
              "2  20345487  1  [[0.26666666666666666, 0.4, 0.2666666666666666...\n",
              "3  23822395  0  [[0.26666666666666666, 0.4, 0.2666666666666666...\n",
              "4  28994087  0  [[0.26666666666666666, 0.4, 0.2666666666666666..."
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-c2dec149-a180-4370-80a9-a1f8fce18743\">\n",
              "    <div class=\"colab-df-container\">\n",
              "      <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>0</th>\n",
              "      <th>1</th>\n",
              "      <th>2</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>26184834</td>\n",
              "      <td>1</td>\n",
              "      <td>[[0.26666666666666666, 0.4, 0.2666666666666666...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>23581541</td>\n",
              "      <td>0</td>\n",
              "      <td>[[0.26666666666666666, 0.4, 0.2666666666666666...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>20345487</td>\n",
              "      <td>1</td>\n",
              "      <td>[[0.26666666666666666, 0.4, 0.2666666666666666...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>23822395</td>\n",
              "      <td>0</td>\n",
              "      <td>[[0.26666666666666666, 0.4, 0.2666666666666666...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>28994087</td>\n",
              "      <td>0</td>\n",
              "      <td>[[0.26666666666666666, 0.4, 0.2666666666666666...</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "</div>\n",
              "      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-c2dec149-a180-4370-80a9-a1f8fce18743')\"\n",
              "              title=\"Convert this dataframe to an interactive table.\"\n",
              "              style=\"display:none;\">\n",
              "        \n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "       width=\"24px\">\n",
              "    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n",
              "    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n",
              "  </svg>\n",
              "      </button>\n",
              "      \n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      flex-wrap:wrap;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "      <script>\n",
              "        const buttonEl =\n",
              "          document.querySelector('#df-c2dec149-a180-4370-80a9-a1f8fce18743 button.colab-df-convert');\n",
              "        buttonEl.style.display =\n",
              "          google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "        async function convertToInteractive(key) {\n",
              "          const element = document.querySelector('#df-c2dec149-a180-4370-80a9-a1f8fce18743');\n",
              "          const dataTable =\n",
              "            await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                     [key], {});\n",
              "          if (!dataTable) return;\n",
              "\n",
              "          const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "            '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "            + ' to learn more about interactive tables.';\n",
              "          element.innerHTML = '';\n",
              "          dataTable['output_type'] = 'display_data';\n",
              "          await google.colab.output.renderOutput(dataTable, element);\n",
              "          const docLink = document.createElement('div');\n",
              "          docLink.innerHTML = docLinkHtml;\n",
              "          element.appendChild(docLink);\n",
              "        }\n",
              "      </script>\n",
              "    </div>\n",
              "  </div>\n",
              "  "
            ]
          },
          "metadata": {},
          "execution_count": 77
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "death_dataset['HADM_IDS'].sort()"
      ],
      "metadata": {
        "id": "V6fMISDJajAy"
      },
      "execution_count": 77,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "len(death_dataset['HADM_IDS'])"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "QuGDgz32arFM",
        "outputId": "5ab0a760-7c46-40f2-acdd-e1387ba4d5a0"
      },
      "execution_count": 80,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "32785"
            ]
          },
          "metadata": {},
          "execution_count": 80
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "len(set(death_dataset['HADM_IDS']))"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "SYugT6x7bI6H",
        "outputId": "1dacc383-9dc5-4f92-eaaa-0f95f53a14b8"
      },
      "execution_count": 82,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "32785"
            ]
          },
          "metadata": {},
          "execution_count": 82
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "## train-test 분할"
      ],
      "metadata": {
        "id": "78GmjAzpqJWt"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "## 학습용 검증용 데이터 나누기 (8:2)\n",
        "def split_train_test(dataset, train_ratio=0.8, pos_neg=True):\n",
        "  #레이블이 양성, 음성인 경우, 학습 데이터, 테스트 데이터를 나눌 때 \n",
        "  #양성, 음성의 비율을 각각 유지할 수 있도록 구현\n",
        "  if pos_neg:\n",
        "    pos_idx = np.where(np.array(dataset['LABEL']) == 1)[0]\n",
        "    neg_idx = np.where(np.array(dataset['LABEL']) == 0)[0]\n",
        "\n",
        "    # suffle 넣어줘서 index 섞어줘야함. np.random, random,suffle\n",
        "    np.random.shuffle(pos_idx)\n",
        "    np.random.shuffle(neg_idx)\n",
        "\n",
        "    pos_num = len(pos_idx)\n",
        "    neg_num = len(neg_idx)\n",
        "\n",
        "    train_pos_idx = list(pos_idx[:int(pos_num * train_ratio)])\n",
        "    test_pos_idx = list(pos_idx[int(pos_num * train_ratio):])\n",
        "\n",
        "    train_neg_idx = list(neg_idx[:int(neg_num * train_ratio)])\n",
        "    test_neg_idx = list(neg_idx[int(neg_num * train_ratio):])\n",
        "\n",
        "    train_idx = train_pos_idx + train_neg_idx\n",
        "    test_idx = test_pos_idx + test_neg_idx\n",
        "  \n",
        "  #레이블이 양성, 음성이 아닌경우 단순하게 전체 데이터에서 학습, 테스트 비율만큼 나누기\n",
        "  else:\n",
        "    idx = np.arange(len(dataset['LABEL']))\n",
        "    np.random.shuffle(idx)\n",
        "    train_idx = idx[:int(len(idx) * train_ratio)]\n",
        "    test_idx = idx[int(len(idx) * train_ratio):]\n",
        "\n",
        "  train_set = dict({\n",
        "      'HADM_IDS': np.array(dataset['HADM_IDS'])[train_idx],\n",
        "      'LABEL': np.array(dataset['LABEL'])[train_idx],\n",
        "      'DATA': np.array(dataset['DATA'])[train_idx]\n",
        "  })\n",
        "\n",
        "  test_set = dict({\n",
        "      'HADM_IDS': np.array(dataset['HADM_IDS'])[test_idx],\n",
        "      'LABEL': np.array(dataset['LABEL'])[test_idx],\n",
        "      'DATA': np.array(dataset['DATA'])[test_idx]\n",
        "  })\n",
        "  return train_set, test_set"
      ],
      "metadata": {
        "id": "o-6aOPFhqFQ_"
      },
      "execution_count": 27,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "death_train, death_test = split_train_test(death_dataset)\n",
        "los_train, los_test = split_train_test(los_dataset, pos_neg=False)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "GI5h60FIqQXK",
        "outputId": "e30668a9-0c1a-4dc7-dec0-54a0f6904eb8"
      },
      "execution_count": 28,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "/usr/local/lib/python3.7/dist-packages/ipykernel_launcher.py:35: VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray.\n",
            "/usr/local/lib/python3.7/dist-packages/ipykernel_launcher.py:41: VisibleDeprecationWarning: Creating an ndarray from ragged nested sequences (which is a list-or-tuple of lists-or-tuples-or ndarrays with different lengths or shapes) is deprecated. If you meant to do this, you must specify 'dtype=object' when creating the ndarray.\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "print(len(death_train['HADM_IDS']))\n",
        "print(len(death_test['HADM_IDS']))\n",
        "print(len(los_train['HADM_IDS']))\n",
        "print(len(los_test['HADM_IDS']))"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "QgYq2iiPqSZj",
        "outputId": "aa873987-56bd-4428-a8ed-0d9e1853241f"
      },
      "execution_count": 29,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "26227\n",
            "6558\n",
            "26072\n",
            "6519\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "len(death_train['HADM_IDS'])/len(death_dataset['HADM_IDS'])"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "ZamPsGztqjuu",
        "outputId": "8003bb29-9e16-4495-8649-20b112eaa1cd"
      },
      "execution_count": 30,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "0.7999694982461492"
            ]
          },
          "metadata": {},
          "execution_count": 30
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "len(los_train['HADM_IDS'])/len(los_dataset['HADM_IDS'])"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "GoDSJBuKq5Rt",
        "outputId": "f08f5a07-8527-4b31-ff84-a0a675963ae4"
      },
      "execution_count": 31,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "0.7999754533460157"
            ]
          },
          "metadata": {},
          "execution_count": 31
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [
        "## 모델 학습 \n"
      ],
      "metadata": {
        "id": "VZvM3xUVqF5A"
      }
    },
    {
      "cell_type": "markdown",
      "source": [],
      "metadata": {
        "id": "3GP_zI7WqFzf"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "import tensorflow as tf\n",
        "from tensorflow import keras\n",
        "from tensorflow.keras import layers\n",
        "import math\n",
        "from tensorflow.keras.utils import Sequence\n",
        "import numpy as np"
      ],
      "metadata": {
        "id": "b38QBa6_TYQm"
      },
      "execution_count": 32,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "## 사망예측 모델용 데이터 로더\n",
        "class Death_Dataloader(Sequence):\n",
        "    def __init__(self, x_set, y_set, hadm_id_set, batch_size, shuffle=False):\n",
        "        self.x, self.y, self.hadm_id = x_set, y_set, hadm_id_set\n",
        "        self.batch_size = batch_size\n",
        "        self.shuffle = shuffle\n",
        "        self.on_epoch_end()\n",
        "\n",
        "    def __len__(self):\n",
        "        return math.ceil(len(self.x) / self.batch_size)\n",
        "\n",
        "\t\t# batch 단위로 직접 묶어줘야 함\n",
        "    def __getitem__(self, idx):\n",
        "\t\t# sampler의 역할(index를 batch_size만큼 sampling해줌)\n",
        "        indices = self.indices[idx*self.batch_size:(idx+1)*self.batch_size]\n",
        "\n",
        "        batch_x = [self.x[i] for i in indices]\n",
        "        ## 입력 데이터들의 크기를 최대 24시간으로 통일하고 24시간 보다 짧은 데이터들의 경우 끝에 '0' padding 붙이기\n",
        "        padded_batch_x = np.zeros((len(batch_x), 24, 16))\n",
        "        for i in range(len(batch_x)):\n",
        "          valid_length = len(batch_x[i])\n",
        "          padded_batch_x[i][:valid_length, :] = batch_x[i]\n",
        "        batch_y = [self.y[i] for i in indices]\n",
        "        batch_hadm_id = [self.hadm_id[i] for i in indices]\n",
        "\n",
        "        return np.array(padded_batch_x).astype('float32'), np.array(batch_y).astype('float32')\n",
        "\n",
        "    # epoch이 끝날때마다 실행\n",
        "    def on_epoch_end(self):\n",
        "        self.indices = np.arange(len(self.x))\n",
        "        if self.shuffle == True:\n",
        "            np.random.shuffle(self.indices)\n",
        "\n",
        "death_train_loader = Death_Dataloader(death_train['DATA'], death_train['LABEL'], death_train['HADM_IDS'], 125, True)\n",
        "death_test_loader = Death_Dataloader(death_test['DATA'], death_test['LABEL'], death_test['HADM_IDS'], 125, False)"
      ],
      "metadata": {
        "id": "jtelKrDCpoOL"
      },
      "execution_count": 54,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "death_train.keys()"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "6d2kHX5YS0Ts",
        "outputId": "f1b0ea99-952f-46f6-f615-b72853c754ca"
      },
      "execution_count": 37,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "dict_keys(['HADM_IDS', 'LABEL', 'DATA'])"
            ]
          },
          "metadata": {},
          "execution_count": 37
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "#death_train['DATA'][0].shape\n",
        "for i in death_train_loader:\n",
        "    print(i)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "c8GVk6HHTTlz",
        "outputId": "9cac7285-ffdc-4c1c-f619-a3c97531160c"
      },
      "execution_count": 57,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "\u001b[1;30;43m스트리밍 출력 내용이 길어서 마지막 5000줄이 삭제되었습니다.\u001b[0m\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.53333336, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.50333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.27      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 1., 1., 0., 1., 0., 0., 1., 0., 1.,\n",
            "       1., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.59      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5233333 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 1.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.28333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.48333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.48333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.48666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.26666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.51666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.53      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.52666664, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.53      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5366667 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.50333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5566667 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5233333 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       1., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 1., 1., 1., 0., 1., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 1., 1., 1., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 1., 1., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 1., 0., 1., 1., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 1., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1.,\n",
            "       0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.51666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.51      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 1., 0., 0., 0., 1., 1., 1., 0., 0., 0., 1., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 1., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 0.,\n",
            "       0., 1., 0., 1., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.24333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5233333 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 1.,\n",
            "       1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0.,\n",
            "       0., 1., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.48      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.54333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 1., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.27333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 1., 1., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 1.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 1., 1., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 1., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.28666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.25666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.25      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ],\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ],\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ]]], dtype=float32), array([0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 1., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       1., 1., 1., 1., 0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 1., 0.,\n",
            "       0., 1., 1., 1., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 1., 0., 0., 1., 0., 0., 0., 1., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 1.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.51      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5133333 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 1., 1., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       1., 0., 1., 0., 0., 1., 1., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 0., 1., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.26      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5933333 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1., 0.,\n",
            "       1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 1., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.50333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.25      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 1., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.50333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.53      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.27      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 1., 0., 0., 1., 1.,\n",
            "       0., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.50666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 1., 0., 1., 0., 0., 0., 1., 0., 1., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 1., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0.,\n",
            "       0., 1., 0., 0., 1., 1., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.27666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 1., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.51      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 1., 1., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 1., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 1., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 1., 0., 0., 1., 1., 0., 0., 0., 1., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.66333336, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.23666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.27      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 1., 0., 1., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.27666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.53      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.60333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 1., 0., 1., 0., 1., 1., 0., 0., 0., 1., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.50333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 1., 1., 1., 1., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1.,\n",
            "       0., 0., 1., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 1., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.52666664, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 1., 1., 0., 1., 0., 1., 0., 0., 0., 1., 1., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 1., 0., 0., 0., 0., 0., 0., 1., 1., 1., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.50666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.46666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 1., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.49666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 1., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 0., 0., 1., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 0., 1., 0., 1., 0., 1., 1., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.28666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 1., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.27333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 1., 1., 1., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 1., 1., 0., 0., 1., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 1., 0., 1., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 1., 0., 1., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.21666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.27      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.27      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.27      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 0., 1., 1., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 1., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 1., 0., 1., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 1.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.50333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 1., 1., 0., 0., 1., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 1., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.50333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 1., 1., 1., 0., 1., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.49      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.53333336, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.48333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0.,\n",
            "       1., 1., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.33      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.53      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.53      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.53      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 1., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.49      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.28333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 1., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 1., 1., 1., 0., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       1., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 1., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.27666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.26666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.37666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.32666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., 1., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.3       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.49333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.52      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.37333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 1., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0.,\n",
            "       0., 0., 1., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 1., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.29666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.32      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.54      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 1., 0., 0., 1., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 1.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 1.,\n",
            "       0., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 1., 1., 0., 1., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ],\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ],\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35666665, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42333335, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 1., 0., 0., 0., 1., 1., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 1., 0., 0., 1., 0., 1., 0., 1., 0., 0., 1., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 1., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0.,\n",
            "       0., 0., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.57      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.51      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.54      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.52666664, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.34333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.35      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.41      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48666668, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.33666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.36333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.46      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.47666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.5133333 , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]]], dtype=float32), array([0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 1., 1.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 0., 0., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 0.], dtype=float32))\n",
            "(array([[[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.45      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.31      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.29333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.30666667, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.44      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38333333, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       ...,\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.36      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.43      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.38666666, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.48333332, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.42      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.39      , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.40333334, 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667]],\n",
            "\n",
            "       [[0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        [0.26666668, 0.4       , 0.26666668, ..., 0.04      ,\n",
            "         0.28      , 0.36666667],\n",
            "        ...,\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ],\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ],\n",
            "        [0.        , 0.        , 0.        , ..., 0.        ,\n",
            "         0.        , 0.        ]]], dtype=float32), array([0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0.,\n",
            "       1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 1., 1., 0., 0.,\n",
            "       0., 0., 0., 0., 1., 0., 1., 0., 1., 0., 0., 0., 0., 0., 1., 0., 0.,\n",
            "       0., 1., 0., 0., 0., 1., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0., 0.,\n",
            "       0., 0., 0., 0., 0., 0., 0., 1., 0., 1., 0., 0., 0., 0., 0., 0., 1.],\n",
            "      dtype=float32))\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "id_cepha = pd.read_csv('/content/gdrive/MyDrive/mimiciv_der/temp/id_cepha.csv', low_memory=False, dtype='str', keep_default_na=False)\n",
        "id_fluoro = pd.read_csv('/content/gdrive/MyDrive/mimiciv_der/temp/id_fluoro.csv', low_memory=False, dtype='str', keep_default_na=False)\n",
        "id_glyco = pd.read_csv('/content/gdrive/MyDrive/mimiciv_der/temp/id_glyco.csv', low_memory=False, dtype='str', keep_default_na=False)\n",
        "id_macro = pd.read_csv('/content/gdrive/MyDrive/mimiciv_der/temp/id_macro.csv', low_memory=False, dtype='str', keep_default_na=False)\n",
        "id_peni = pd.read_csv('/content/gdrive/MyDrive/mimiciv_der/temp/id_peni.csv', low_memory=False, dtype='str', keep_default_na=False)\n"
      ],
      "metadata": {
        "id": "8fQJzXzkUnl5"
      },
      "execution_count": 61,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
        "\n",
        "diagnoses_icd = pd.merge(left = diagnoses_icd, right = sepsis[['subject_id','sepsis3']], how='inner', left_on='subject_id', right_on='subject_id')\n"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/",
          "height": 424
        },
        "id": "lvP-e_o2XmpP",
        "outputId": "7a0826d6-1239-4aa2-c486-53613a69c5f8"
      },
      "execution_count": 64,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "        hadm_id   stay_id\n",
              "0      23836605  30001947\n",
              "1      28841024  30002415\n",
              "2      24622512  30002548\n",
              "3      20012928  30003226\n",
              "4      24753602  30003275\n",
              "...         ...       ...\n",
              "15815  27586853  39994129\n",
              "15816  29577504  39996073\n",
              "15817  27161460  39999384\n",
              "15818  26910235  39999552\n",
              "15819  25335698  39999562\n",
              "\n",
              "[15820 rows x 2 columns]"
            ],
            "text/html": [
              "\n",
              "  <div id=\"df-154a3aa0-8324-4c9e-93b0-c3c87a0f6c9f\">\n",
              "    <div class=\"colab-df-container\">\n",
              "      <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>hadm_id</th>\n",
              "      <th>stay_id</th>\n",
              "    </tr>\n",
              "  </thead>\n",
              "  <tbody>\n",
              "    <tr>\n",
              "      <th>0</th>\n",
              "      <td>23836605</td>\n",
              "      <td>30001947</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>1</th>\n",
              "      <td>28841024</td>\n",
              "      <td>30002415</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>2</th>\n",
              "      <td>24622512</td>\n",
              "      <td>30002548</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>3</th>\n",
              "      <td>20012928</td>\n",
              "      <td>30003226</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>4</th>\n",
              "      <td>24753602</td>\n",
              "      <td>30003275</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>...</th>\n",
              "      <td>...</td>\n",
              "      <td>...</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>15815</th>\n",
              "      <td>27586853</td>\n",
              "      <td>39994129</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>15816</th>\n",
              "      <td>29577504</td>\n",
              "      <td>39996073</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>15817</th>\n",
              "      <td>27161460</td>\n",
              "      <td>39999384</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>15818</th>\n",
              "      <td>26910235</td>\n",
              "      <td>39999552</td>\n",
              "    </tr>\n",
              "    <tr>\n",
              "      <th>15819</th>\n",
              "      <td>25335698</td>\n",
              "      <td>39999562</td>\n",
              "    </tr>\n",
              "  </tbody>\n",
              "</table>\n",
              "<p>15820 rows × 2 columns</p>\n",
              "</div>\n",
              "      <button class=\"colab-df-convert\" onclick=\"convertToInteractive('df-154a3aa0-8324-4c9e-93b0-c3c87a0f6c9f')\"\n",
              "              title=\"Convert this dataframe to an interactive table.\"\n",
              "              style=\"display:none;\">\n",
              "        \n",
              "  <svg xmlns=\"http://www.w3.org/2000/svg\" height=\"24px\"viewBox=\"0 0 24 24\"\n",
              "       width=\"24px\">\n",
              "    <path d=\"M0 0h24v24H0V0z\" fill=\"none\"/>\n",
              "    <path d=\"M18.56 5.44l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94zm-11 1L8.5 8.5l.94-2.06 2.06-.94-2.06-.94L8.5 2.5l-.94 2.06-2.06.94zm10 10l.94 2.06.94-2.06 2.06-.94-2.06-.94-.94-2.06-.94 2.06-2.06.94z\"/><path d=\"M17.41 7.96l-1.37-1.37c-.4-.4-.92-.59-1.43-.59-.52 0-1.04.2-1.43.59L10.3 9.45l-7.72 7.72c-.78.78-.78 2.05 0 2.83L4 21.41c.39.39.9.59 1.41.59.51 0 1.02-.2 1.41-.59l7.78-7.78 2.81-2.81c.8-.78.8-2.07 0-2.86zM5.41 20L4 18.59l7.72-7.72 1.47 1.35L5.41 20z\"/>\n",
              "  </svg>\n",
              "      </button>\n",
              "      \n",
              "  <style>\n",
              "    .colab-df-container {\n",
              "      display:flex;\n",
              "      flex-wrap:wrap;\n",
              "      gap: 12px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert {\n",
              "      background-color: #E8F0FE;\n",
              "      border: none;\n",
              "      border-radius: 50%;\n",
              "      cursor: pointer;\n",
              "      display: none;\n",
              "      fill: #1967D2;\n",
              "      height: 32px;\n",
              "      padding: 0 0 0 0;\n",
              "      width: 32px;\n",
              "    }\n",
              "\n",
              "    .colab-df-convert:hover {\n",
              "      background-color: #E2EBFA;\n",
              "      box-shadow: 0px 1px 2px rgba(60, 64, 67, 0.3), 0px 1px 3px 1px rgba(60, 64, 67, 0.15);\n",
              "      fill: #174EA6;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert {\n",
              "      background-color: #3B4455;\n",
              "      fill: #D2E3FC;\n",
              "    }\n",
              "\n",
              "    [theme=dark] .colab-df-convert:hover {\n",
              "      background-color: #434B5C;\n",
              "      box-shadow: 0px 1px 3px 1px rgba(0, 0, 0, 0.15);\n",
              "      filter: drop-shadow(0px 1px 2px rgba(0, 0, 0, 0.3));\n",
              "      fill: #FFFFFF;\n",
              "    }\n",
              "  </style>\n",
              "\n",
              "      <script>\n",
              "        const buttonEl =\n",
              "          document.querySelector('#df-154a3aa0-8324-4c9e-93b0-c3c87a0f6c9f button.colab-df-convert');\n",
              "        buttonEl.style.display =\n",
              "          google.colab.kernel.accessAllowed ? 'block' : 'none';\n",
              "\n",
              "        async function convertToInteractive(key) {\n",
              "          const element = document.querySelector('#df-154a3aa0-8324-4c9e-93b0-c3c87a0f6c9f');\n",
              "          const dataTable =\n",
              "            await google.colab.kernel.invokeFunction('convertToInteractive',\n",
              "                                                     [key], {});\n",
              "          if (!dataTable) return;\n",
              "\n",
              "          const docLinkHtml = 'Like what you see? Visit the ' +\n",
              "            '<a target=\"_blank\" href=https://colab.research.google.com/notebooks/data_table.ipynb>data table notebook</a>'\n",
              "            + ' to learn more about interactive tables.';\n",
              "          element.innerHTML = '';\n",
              "          dataTable['output_type'] = 'display_data';\n",
              "          await google.colab.output.renderOutput(dataTable, element);\n",
              "          const docLink = document.createElement('div');\n",
              "          docLink.innerHTML = docLinkHtml;\n",
              "          element.appendChild(docLink);\n",
              "        }\n",
              "      </script>\n",
              "    </div>\n",
              "  </div>\n",
              "  "
            ]
          },
          "metadata": {},
          "execution_count": 64
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "## 사망예측 모델\n",
        "death_lstm = keras.Sequential([\n",
        "    layers.LSTM(25, input_shape=(13,16)),\n",
        "    layers.Dense(125, activation='relu'),\n",
        "    layers.Dense(1, activation='sigmoid')\n",
        "])\n",
        "death_lstm.summary()"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "bn7zfO5e_kRw",
        "outputId": "187e22db-03c2-47d7-83db-9964fb5a54ed"
      },
      "execution_count": 58,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Model: \"sequential_2\"\n",
            "_________________________________________________________________\n",
            " Layer (type)                Output Shape              Param #   \n",
            "=================================================================\n",
            " lstm_2 (LSTM)               (None, 25)                4200      \n",
            "                                                                 \n",
            " dense_4 (Dense)             (None, 125)               3250      \n",
            "                                                                 \n",
            " dense_5 (Dense)             (None, 1)                 126       \n",
            "                                                                 \n",
            "=================================================================\n",
            "Total params: 7,576\n",
            "Trainable params: 7,576\n",
            "Non-trainable params: 0\n",
            "_________________________________________________________________\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "## 사망예측 모델 학습\n",
        "death_lstm.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])\n",
        "num_epochs = 50\n",
        "history = death_lstm.fit(death_train_loader, validation_data=death_test_loader, epochs=num_epochs, verbose=2)"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "N7BtKf5g_t7c",
        "outputId": "a5458e80-e3cc-4bc9-f4e8-6e1f30981b14"
      },
      "execution_count": 59,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "Epoch 1/50\n",
            "210/210 - 10s - loss: 0.4405 - accuracy: 0.8325 - val_loss: 0.4190 - val_accuracy: 0.8521 - 10s/epoch - 47ms/step\n",
            "Epoch 2/50\n",
            "210/210 - 1s - loss: 0.4156 - accuracy: 0.8516 - val_loss: 0.4186 - val_accuracy: 0.8518 - 1s/epoch - 5ms/step\n",
            "Epoch 3/50\n",
            "210/210 - 1s - loss: 0.4144 - accuracy: 0.8515 - val_loss: 0.4156 - val_accuracy: 0.8521 - 1s/epoch - 5ms/step\n",
            "Epoch 4/50\n",
            "210/210 - 1s - loss: 0.4127 - accuracy: 0.8516 - val_loss: 0.4133 - val_accuracy: 0.8518 - 1s/epoch - 5ms/step\n",
            "Epoch 5/50\n",
            "210/210 - 1s - loss: 0.4106 - accuracy: 0.8516 - val_loss: 0.4149 - val_accuracy: 0.8518 - 1s/epoch - 5ms/step\n",
            "Epoch 6/50\n",
            "210/210 - 1s - loss: 0.4105 - accuracy: 0.8518 - val_loss: 0.4130 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 7/50\n",
            "210/210 - 1s - loss: 0.4109 - accuracy: 0.8518 - val_loss: 0.4161 - val_accuracy: 0.8518 - 1s/epoch - 5ms/step\n",
            "Epoch 8/50\n",
            "210/210 - 1s - loss: 0.4094 - accuracy: 0.8521 - val_loss: 0.4114 - val_accuracy: 0.8521 - 1s/epoch - 5ms/step\n",
            "Epoch 9/50\n",
            "210/210 - 1s - loss: 0.4089 - accuracy: 0.8518 - val_loss: 0.4133 - val_accuracy: 0.8519 - 1s/epoch - 5ms/step\n",
            "Epoch 10/50\n",
            "210/210 - 1s - loss: 0.4088 - accuracy: 0.8519 - val_loss: 0.4130 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 11/50\n",
            "210/210 - 1s - loss: 0.4083 - accuracy: 0.8516 - val_loss: 0.4120 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 12/50\n",
            "210/210 - 1s - loss: 0.4079 - accuracy: 0.8525 - val_loss: 0.4117 - val_accuracy: 0.8519 - 1s/epoch - 5ms/step\n",
            "Epoch 13/50\n",
            "210/210 - 1s - loss: 0.4083 - accuracy: 0.8519 - val_loss: 0.4174 - val_accuracy: 0.8471 - 1s/epoch - 5ms/step\n",
            "Epoch 14/50\n",
            "210/210 - 1s - loss: 0.4075 - accuracy: 0.8516 - val_loss: 0.4118 - val_accuracy: 0.8518 - 1s/epoch - 5ms/step\n",
            "Epoch 15/50\n",
            "210/210 - 1s - loss: 0.4080 - accuracy: 0.8518 - val_loss: 0.4142 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 16/50\n",
            "210/210 - 1s - loss: 0.4073 - accuracy: 0.8520 - val_loss: 0.4127 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 17/50\n",
            "210/210 - 1s - loss: 0.4069 - accuracy: 0.8523 - val_loss: 0.4100 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 18/50\n",
            "210/210 - 1s - loss: 0.4075 - accuracy: 0.8522 - val_loss: 0.4124 - val_accuracy: 0.8513 - 1s/epoch - 5ms/step\n",
            "Epoch 19/50\n",
            "210/210 - 1s - loss: 0.4073 - accuracy: 0.8523 - val_loss: 0.4094 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 20/50\n",
            "210/210 - 1s - loss: 0.4063 - accuracy: 0.8524 - val_loss: 0.4087 - val_accuracy: 0.8529 - 1s/epoch - 5ms/step\n",
            "Epoch 21/50\n",
            "210/210 - 1s - loss: 0.4060 - accuracy: 0.8523 - val_loss: 0.4083 - val_accuracy: 0.8516 - 1s/epoch - 5ms/step\n",
            "Epoch 22/50\n",
            "210/210 - 1s - loss: 0.4071 - accuracy: 0.8519 - val_loss: 0.4089 - val_accuracy: 0.8529 - 1s/epoch - 5ms/step\n",
            "Epoch 23/50\n",
            "210/210 - 1s - loss: 0.4064 - accuracy: 0.8525 - val_loss: 0.4097 - val_accuracy: 0.8519 - 1s/epoch - 5ms/step\n",
            "Epoch 24/50\n",
            "210/210 - 1s - loss: 0.4069 - accuracy: 0.8526 - val_loss: 0.4093 - val_accuracy: 0.8527 - 1s/epoch - 5ms/step\n",
            "Epoch 25/50\n",
            "210/210 - 1s - loss: 0.4063 - accuracy: 0.8524 - val_loss: 0.4117 - val_accuracy: 0.8513 - 1s/epoch - 5ms/step\n",
            "Epoch 26/50\n",
            "210/210 - 1s - loss: 0.4061 - accuracy: 0.8529 - val_loss: 0.4091 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 27/50\n",
            "210/210 - 1s - loss: 0.4062 - accuracy: 0.8526 - val_loss: 0.4095 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 28/50\n",
            "210/210 - 1s - loss: 0.4059 - accuracy: 0.8529 - val_loss: 0.4085 - val_accuracy: 0.8529 - 1s/epoch - 5ms/step\n",
            "Epoch 29/50\n",
            "210/210 - 1s - loss: 0.4058 - accuracy: 0.8527 - val_loss: 0.4148 - val_accuracy: 0.8503 - 1s/epoch - 5ms/step\n",
            "Epoch 30/50\n",
            "210/210 - 1s - loss: 0.4059 - accuracy: 0.8523 - val_loss: 0.4070 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 31/50\n",
            "210/210 - 1s - loss: 0.4064 - accuracy: 0.8525 - val_loss: 0.4082 - val_accuracy: 0.8525 - 1s/epoch - 5ms/step\n",
            "Epoch 32/50\n",
            "210/210 - 1s - loss: 0.4055 - accuracy: 0.8525 - val_loss: 0.4095 - val_accuracy: 0.8530 - 1s/epoch - 5ms/step\n",
            "Epoch 33/50\n",
            "210/210 - 1s - loss: 0.4055 - accuracy: 0.8528 - val_loss: 0.4085 - val_accuracy: 0.8522 - 1s/epoch - 5ms/step\n",
            "Epoch 34/50\n",
            "210/210 - 1s - loss: 0.4045 - accuracy: 0.8528 - val_loss: 0.4073 - val_accuracy: 0.8535 - 1s/epoch - 5ms/step\n",
            "Epoch 35/50\n",
            "210/210 - 1s - loss: 0.4053 - accuracy: 0.8527 - val_loss: 0.4086 - val_accuracy: 0.8530 - 1s/epoch - 5ms/step\n",
            "Epoch 36/50\n",
            "210/210 - 1s - loss: 0.4052 - accuracy: 0.8529 - val_loss: 0.4081 - val_accuracy: 0.8524 - 1s/epoch - 5ms/step\n",
            "Epoch 37/50\n",
            "210/210 - 1s - loss: 0.4055 - accuracy: 0.8527 - val_loss: 0.4089 - val_accuracy: 0.8529 - 1s/epoch - 5ms/step\n",
            "Epoch 38/50\n",
            "210/210 - 1s - loss: 0.4050 - accuracy: 0.8531 - val_loss: 0.4075 - val_accuracy: 0.8533 - 1s/epoch - 5ms/step\n",
            "Epoch 39/50\n",
            "210/210 - 1s - loss: 0.4049 - accuracy: 0.8529 - val_loss: 0.4075 - val_accuracy: 0.8532 - 1s/epoch - 5ms/step\n",
            "Epoch 40/50\n",
            "210/210 - 1s - loss: 0.4049 - accuracy: 0.8528 - val_loss: 0.4070 - val_accuracy: 0.8532 - 1s/epoch - 5ms/step\n",
            "Epoch 41/50\n",
            "210/210 - 1s - loss: 0.4049 - accuracy: 0.8530 - val_loss: 0.4061 - val_accuracy: 0.8524 - 1s/epoch - 5ms/step\n",
            "Epoch 42/50\n",
            "210/210 - 1s - loss: 0.4047 - accuracy: 0.8537 - val_loss: 0.4063 - val_accuracy: 0.8535 - 1s/epoch - 5ms/step\n",
            "Epoch 43/50\n",
            "210/210 - 1s - loss: 0.4036 - accuracy: 0.8531 - val_loss: 0.4078 - val_accuracy: 0.8527 - 1s/epoch - 5ms/step\n",
            "Epoch 44/50\n",
            "210/210 - 1s - loss: 0.4045 - accuracy: 0.8526 - val_loss: 0.4101 - val_accuracy: 0.8516 - 1s/epoch - 5ms/step\n",
            "Epoch 45/50\n",
            "210/210 - 1s - loss: 0.4039 - accuracy: 0.8534 - val_loss: 0.4076 - val_accuracy: 0.8521 - 1s/epoch - 5ms/step\n",
            "Epoch 46/50\n",
            "210/210 - 1s - loss: 0.4039 - accuracy: 0.8531 - val_loss: 0.4086 - val_accuracy: 0.8530 - 1s/epoch - 5ms/step\n",
            "Epoch 47/50\n",
            "210/210 - 1s - loss: 0.4040 - accuracy: 0.8531 - val_loss: 0.4080 - val_accuracy: 0.8532 - 1s/epoch - 5ms/step\n",
            "Epoch 48/50\n",
            "210/210 - 1s - loss: 0.4052 - accuracy: 0.8531 - val_loss: 0.4073 - val_accuracy: 0.8529 - 1s/epoch - 5ms/step\n",
            "Epoch 49/50\n",
            "210/210 - 1s - loss: 0.4041 - accuracy: 0.8534 - val_loss: 0.4057 - val_accuracy: 0.8529 - 1s/epoch - 5ms/step\n",
            "Epoch 50/50\n",
            "210/210 - 1s - loss: 0.4032 - accuracy: 0.8533 - val_loss: 0.4069 - val_accuracy: 0.8524 - 1s/epoch - 5ms/step\n"
          ]
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [],
      "metadata": {
        "id": "ABTMAjyBXfd-"
      }
    },
    {
      "cell_type": "code",
      "source": [
        "sum(death_train['LABEL'])/len(death_train['LABEL'])"
      ],
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "id": "lWCgxDDqYNn2",
        "outputId": "ac28ff9b-658a-4ee7-fb00-0c4bac97473c"
      },
      "execution_count": 63,
      "outputs": [
        {
          "output_type": "execute_result",
          "data": {
            "text/plain": [
              "0.15838639569908872"
            ]
          },
          "metadata": {},
          "execution_count": 63
        }
      ]
    },
    {
      "cell_type": "markdown",
      "source": [],
      "metadata": {
        "id": "Bp43b-ctXfbj"
      }
    }
  ],
  "metadata": {
    "colab": {
      "collapsed_sections": [],
      "provenance": [],
      "machine_shape": "hm",
      "include_colab_link": true
    },
    "kernelspec": {
      "display_name": "Python 3",
      "name": "python3"
    },
    "language_info": {
      "name": "python"
    },
    "widgets": {
      "application/vnd.jupyter.widget-state+json": {
        "5472ca8f6e3a403f91739d19b8faea98": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_976b2e903f2049bf927d87eb31eafa31",
              "IPY_MODEL_68643b54cbc74e93a8348379aa6652dc",
              "IPY_MODEL_79e802f0d027404cb608633f20c01207"
            ],
            "layout": "IPY_MODEL_6c20ab6bf17f4ea0b8f7b3a60352fe59"
          }
        },
        "976b2e903f2049bf927d87eb31eafa31": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_3b0066087ebf403fb7553faed14e3cad",
            "placeholder": "​",
            "style": "IPY_MODEL_0f552905a4f74370959d443c767997a1",
            "value": "Extract_VS: "
          }
        },
        "68643b54cbc74e93a8348379aa6652dc": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_fc4da944755b4bec9cc17810bd45cdcc",
            "max": 7241351,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_648fd305dd104142b305433c56798fae",
            "value": 7241351
          }
        },
        "79e802f0d027404cb608633f20c01207": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_2403faf1c0cc4ef8bcfcd64e99b255fa",
            "placeholder": "​",
            "style": "IPY_MODEL_eabd8a510ecc48f9a494452885a3de3d",
            "value": " 329822285/? [37:34&lt;00:00, 146745.67it/s]"
          }
        },
        "6c20ab6bf17f4ea0b8f7b3a60352fe59": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": "inline-flex",
            "flex": null,
            "flex_flow": "row wrap",
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": "700px"
          }
        },
        "3b0066087ebf403fb7553faed14e3cad": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "0f552905a4f74370959d443c767997a1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "fc4da944755b4bec9cc17810bd45cdcc": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": "2",
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "648fd305dd104142b305433c56798fae": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "2403faf1c0cc4ef8bcfcd64e99b255fa": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "eabd8a510ecc48f9a494452885a3de3d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "f326faeebfc84cf6baeb0f1e35cb29af": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_2036c48bed07415a9f6d733d1899cf04",
              "IPY_MODEL_3f62d85a1c2d4031bf9f1b3a80f454e7",
              "IPY_MODEL_8cfee1ca14a84ae88d5d47d6c2a4874f"
            ],
            "layout": "IPY_MODEL_105ad6d8432b483d8ee44824a89e3603"
          }
        },
        "2036c48bed07415a9f6d733d1899cf04": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_24c16358d2894ab191fe8d976fec2b8f",
            "placeholder": "​",
            "style": "IPY_MODEL_8fa7f9fec3514301ab191875dcd17ae0",
            "value": "Extract_LAB: "
          }
        },
        "3f62d85a1c2d4031bf9f1b3a80f454e7": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_5e64c2da6e464fa6871b7e2594c55b27",
            "max": 615328,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_bdfd3e40c1f94232bba5b112cd5b94ce",
            "value": 615328
          }
        },
        "8cfee1ca14a84ae88d5d47d6c2a4874f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_f59ec96981994f8288e939c4bf411f99",
            "placeholder": "​",
            "style": "IPY_MODEL_6882ecbd9730474dad27c0a7888cf7df",
            "value": " 124342638/? [13:33&lt;00:00, 152191.29it/s]"
          }
        },
        "105ad6d8432b483d8ee44824a89e3603": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": "inline-flex",
            "flex": null,
            "flex_flow": "row wrap",
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": "700px"
          }
        },
        "24c16358d2894ab191fe8d976fec2b8f": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "8fa7f9fec3514301ab191875dcd17ae0": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "5e64c2da6e464fa6871b7e2594c55b27": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": "2",
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "bdfd3e40c1f94232bba5b112cd5b94ce": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "f59ec96981994f8288e939c4bf411f99": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "6882ecbd9730474dad27c0a7888cf7df": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "99a409963f2e4bd28f8b6cb935971a90": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_aab2360df1ef43eb85eba7c663aa37eb",
              "IPY_MODEL_30be9461262e4a8b832c5c7d92891256",
              "IPY_MODEL_1627cc744fde4d9fbe3a33efaa8834ec"
            ],
            "layout": "IPY_MODEL_6b20b1dc24ee4fa0944cb80e8727ed5a"
          }
        },
        "aab2360df1ef43eb85eba7c663aa37eb": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_c34cebd5d4e34e318dcbfdae81d7e86d",
            "placeholder": "​",
            "style": "IPY_MODEL_20856359ee754f51a5a03d59dc078815",
            "value": "Get Death Label: "
          }
        },
        "30be9461262e4a8b832c5c7d92891256": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_3e6d9b592e5e41a8aaf7620ee4fe84c5",
            "max": 1,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_5fecb08f5f5a43018377ca399bc57f65",
            "value": 1
          }
        },
        "1627cc744fde4d9fbe3a33efaa8834ec": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_efa07aac08a14f5a89ba4a2e2b8b3071",
            "placeholder": "​",
            "style": "IPY_MODEL_6f4c931c45d54eccb26f5a64ccbd321c",
            "value": " 195836/? [00:07&lt;00:00, 28835.59it/s]"
          }
        },
        "6b20b1dc24ee4fa0944cb80e8727ed5a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "c34cebd5d4e34e318dcbfdae81d7e86d": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "20856359ee754f51a5a03d59dc078815": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "3e6d9b592e5e41a8aaf7620ee4fe84c5": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": "20px"
          }
        },
        "5fecb08f5f5a43018377ca399bc57f65": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "efa07aac08a14f5a89ba4a2e2b8b3071": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "6f4c931c45d54eccb26f5a64ccbd321c": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "3a814315b42c43f4a59cd571186189f6": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_906c72d94c1d495bbad9ed08917f9b83",
              "IPY_MODEL_15f42cc8f32c4dd198e255c8e6b78217",
              "IPY_MODEL_3bcc410f5b21483fa345c4cf1f2f0af4"
            ],
            "layout": "IPY_MODEL_f6437692ca0d4c049423f80706fc24a6"
          }
        },
        "906c72d94c1d495bbad9ed08917f9b83": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_c16ca63c7851488eb13d26f99058b86c",
            "placeholder": "​",
            "style": "IPY_MODEL_94a2168440b14a978d5dc17a309c06c8",
            "value": "Get loss Label: "
          }
        },
        "15f42cc8f32c4dd198e255c8e6b78217": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_4fe63e54e192470b91014fb139d84e63",
            "max": 1,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_77e4e875434a4d54adedc02ea6630ce4",
            "value": 1
          }
        },
        "3bcc410f5b21483fa345c4cf1f2f0af4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_f033a915a1fe40b2afbcf593694c3e30",
            "placeholder": "​",
            "style": "IPY_MODEL_fbc98104317d4d40947e6c4cb7561109",
            "value": " 195836/? [00:11&lt;00:00, 16332.99it/s]"
          }
        },
        "f6437692ca0d4c049423f80706fc24a6": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "c16ca63c7851488eb13d26f99058b86c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "94a2168440b14a978d5dc17a309c06c8": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "4fe63e54e192470b91014fb139d84e63": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": "20px"
          }
        },
        "77e4e875434a4d54adedc02ea6630ce4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "f033a915a1fe40b2afbcf593694c3e30": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "fbc98104317d4d40947e6c4cb7561109": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "9debacc05af94edbaee6d215adce9028": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_c43951b1593c4b69af720286ceacf623",
              "IPY_MODEL_47ee97ea6f424167a79ae31e3b084e2b",
              "IPY_MODEL_717dd6e408294aa995460682eb8bda6e"
            ],
            "layout": "IPY_MODEL_fb9ae9afd01942c48ea79f999629943e"
          }
        },
        "c43951b1593c4b69af720286ceacf623": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_daed62cfa52945a5b8ba2d4729f7d467",
            "placeholder": "​",
            "style": "IPY_MODEL_867f5717f32442eca389a33478203f7b",
            "value": "Doing Quantization: 100%"
          }
        },
        "47ee97ea6f424167a79ae31e3b084e2b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_8560981b4fe544018079359126904fb1",
            "max": 32812,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_2747112928424905afdc9f1af291e344",
            "value": 32812
          }
        },
        "717dd6e408294aa995460682eb8bda6e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_29067893497e418b848a0055ac5049d3",
            "placeholder": "​",
            "style": "IPY_MODEL_54c37704080346ffbceaaa08cbab6e7d",
            "value": " 32812/32812 [00:41&lt;00:00, 832.79it/s]"
          }
        },
        "fb9ae9afd01942c48ea79f999629943e": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "daed62cfa52945a5b8ba2d4729f7d467": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "867f5717f32442eca389a33478203f7b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "8560981b4fe544018079359126904fb1": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "2747112928424905afdc9f1af291e344": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "29067893497e418b848a0055ac5049d3": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "54c37704080346ffbceaaa08cbab6e7d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "eae23d3ecca048dfbe0972630280425e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_b1338e8b13d340bfbd5923558b12d9af",
              "IPY_MODEL_906bb79f38314401b11ae2932482d53a",
              "IPY_MODEL_832c89ff71714c0b96187a45020041c4"
            ],
            "layout": "IPY_MODEL_387570612c284612888c0e020f28cbf6"
          }
        },
        "b1338e8b13d340bfbd5923558b12d9af": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_7a8175aa904d40b8973239cca8b9d530",
            "placeholder": "​",
            "style": "IPY_MODEL_79d42ecf68984391b49b58c4238085de",
            "value": "Carry-Forward: 100%"
          }
        },
        "906bb79f38314401b11ae2932482d53a": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_57d69c105ef44ecdaed386ac4063697d",
            "max": 32785,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_837fbb0216e74cf889e32af4a77d9dc9",
            "value": 32785
          }
        },
        "832c89ff71714c0b96187a45020041c4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_35aa6271461d4529b669e4327b6433ef",
            "placeholder": "​",
            "style": "IPY_MODEL_01ca384bcbb145f1885c96da3eb83b82",
            "value": " 32785/32785 [04:47&lt;00:00, 115.32it/s]"
          }
        },
        "387570612c284612888c0e020f28cbf6": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "7a8175aa904d40b8973239cca8b9d530": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "79d42ecf68984391b49b58c4238085de": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "57d69c105ef44ecdaed386ac4063697d": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "837fbb0216e74cf889e32af4a77d9dc9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "35aa6271461d4529b669e4327b6433ef": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "01ca384bcbb145f1885c96da3eb83b82": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "fffe02e0eacb404580aa1de153e0a157": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_7bc518e6d26146c98226a28a2a7363ef",
              "IPY_MODEL_dba7241eee804b56b52f010b75401fc1",
              "IPY_MODEL_276fc4ae40f64a1696b2843e5e10c116"
            ],
            "layout": "IPY_MODEL_c765eb6b4423459c8eac49d006c50623"
          }
        },
        "7bc518e6d26146c98226a28a2a7363ef": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_a24e7ccb9b1c4659ab6453c5e854631c",
            "placeholder": "​",
            "style": "IPY_MODEL_e41cedbe0cd443efb48c5d3c2e107ce5",
            "value": "Min-Max Normalization: 100%"
          }
        },
        "dba7241eee804b56b52f010b75401fc1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_e44e6718c2ee4616b6aa380e03716466",
            "max": 32785,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_98004a3bdd344ef09983c8ffe04298af",
            "value": 32785
          }
        },
        "276fc4ae40f64a1696b2843e5e10c116": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_ef586eeca984410f8778eac0f736b40c",
            "placeholder": "​",
            "style": "IPY_MODEL_e266f2c4a2fc4aa59686397701a72695",
            "value": " 32785/32785 [00:26&lt;00:00, 1231.97it/s]"
          }
        },
        "c765eb6b4423459c8eac49d006c50623": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": "inline-flex",
            "flex": null,
            "flex_flow": "row wrap",
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": "700px"
          }
        },
        "a24e7ccb9b1c4659ab6453c5e854631c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "e41cedbe0cd443efb48c5d3c2e107ce5": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "e44e6718c2ee4616b6aa380e03716466": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": "2",
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "98004a3bdd344ef09983c8ffe04298af": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "ef586eeca984410f8778eac0f736b40c": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "e266f2c4a2fc4aa59686397701a72695": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "716537a7fcb2471e9857bba28c484cdd": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_a5a2f326cf9946d89bddeaf0ccb714b1",
              "IPY_MODEL_62681f4b05e74186b91489e682150fe4",
              "IPY_MODEL_e870357a5c7a41db89ffa596e99f6b3d"
            ],
            "layout": "IPY_MODEL_fea6c742f7dd46ea8836080c90079aaa"
          }
        },
        "a5a2f326cf9946d89bddeaf0ccb714b1": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_5ce939731df9495bb4c430157f35a2d5",
            "placeholder": "​",
            "style": "IPY_MODEL_650a30f0863746318784885af11cef7b",
            "value": "Make Death Dataset: 100%"
          }
        },
        "62681f4b05e74186b91489e682150fe4": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_4de03ebccf714493b1d0fcae20bfdc8b",
            "max": 107935,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_91a9e58cbbbf4b3aae6e4e161713b3b9",
            "value": 107935
          }
        },
        "e870357a5c7a41db89ffa596e99f6b3d": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_b99411094dc5458d850b8aa17649e2e5",
            "placeholder": "​",
            "style": "IPY_MODEL_6341951516234d30a3a9dd900856db6f",
            "value": " 107935/107935 [00:02&lt;00:00, 47496.90it/s]"
          }
        },
        "fea6c742f7dd46ea8836080c90079aaa": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": "inline-flex",
            "flex": null,
            "flex_flow": "row wrap",
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": "700px"
          }
        },
        "5ce939731df9495bb4c430157f35a2d5": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "650a30f0863746318784885af11cef7b": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "4de03ebccf714493b1d0fcae20bfdc8b": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": "2",
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "91a9e58cbbbf4b3aae6e4e161713b3b9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "b99411094dc5458d850b8aa17649e2e5": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "6341951516234d30a3a9dd900856db6f": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "120ca2b8d38f411496c6bb72b653d090": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HBoxModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HBoxModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HBoxView",
            "box_style": "",
            "children": [
              "IPY_MODEL_7af63be1ccd14f708b4b77f2fbb38b26",
              "IPY_MODEL_c72a6d8b5cbf4c70a25bf1330c872d2e",
              "IPY_MODEL_9041ccad9bb24822be512ba6e582ef09"
            ],
            "layout": "IPY_MODEL_f0483655f18943e6a1631b62395b704a"
          }
        },
        "7af63be1ccd14f708b4b77f2fbb38b26": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_af624fee48934f2a9c4001b27ad5faa7",
            "placeholder": "​",
            "style": "IPY_MODEL_2cfabc724ae64c2ab78a2383f47ec6a9",
            "value": "Make LOS Dataset: 100%"
          }
        },
        "c72a6d8b5cbf4c70a25bf1330c872d2e": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "FloatProgressModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "FloatProgressModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "ProgressView",
            "bar_style": "success",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_6c507a638fb84705b680d22db7b20d27",
            "max": 107935,
            "min": 0,
            "orientation": "horizontal",
            "style": "IPY_MODEL_f7bb27ca58704a0c93f03e31ae1dcddd",
            "value": 107935
          }
        },
        "9041ccad9bb24822be512ba6e582ef09": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "HTMLModel",
          "model_module_version": "1.5.0",
          "state": {
            "_dom_classes": [],
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "HTMLModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/controls",
            "_view_module_version": "1.5.0",
            "_view_name": "HTMLView",
            "description": "",
            "description_tooltip": null,
            "layout": "IPY_MODEL_07194774c7d541eea5e749d98dce11c3",
            "placeholder": "​",
            "style": "IPY_MODEL_89ebf360b3f74a17829e4cf762d4dfbd",
            "value": " 107935/107935 [00:02&lt;00:00, 53017.53it/s]"
          }
        },
        "f0483655f18943e6a1631b62395b704a": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": "inline-flex",
            "flex": null,
            "flex_flow": "row wrap",
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": "700px"
          }
        },
        "af624fee48934f2a9c4001b27ad5faa7": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "2cfabc724ae64c2ab78a2383f47ec6a9": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        },
        "6c507a638fb84705b680d22db7b20d27": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": "2",
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "f7bb27ca58704a0c93f03e31ae1dcddd": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "ProgressStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "ProgressStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "bar_color": null,
            "description_width": ""
          }
        },
        "07194774c7d541eea5e749d98dce11c3": {
          "model_module": "@jupyter-widgets/base",
          "model_name": "LayoutModel",
          "model_module_version": "1.2.0",
          "state": {
            "_model_module": "@jupyter-widgets/base",
            "_model_module_version": "1.2.0",
            "_model_name": "LayoutModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "LayoutView",
            "align_content": null,
            "align_items": null,
            "align_self": null,
            "border": null,
            "bottom": null,
            "display": null,
            "flex": null,
            "flex_flow": null,
            "grid_area": null,
            "grid_auto_columns": null,
            "grid_auto_flow": null,
            "grid_auto_rows": null,
            "grid_column": null,
            "grid_gap": null,
            "grid_row": null,
            "grid_template_areas": null,
            "grid_template_columns": null,
            "grid_template_rows": null,
            "height": null,
            "justify_content": null,
            "justify_items": null,
            "left": null,
            "margin": null,
            "max_height": null,
            "max_width": null,
            "min_height": null,
            "min_width": null,
            "object_fit": null,
            "object_position": null,
            "order": null,
            "overflow": null,
            "overflow_x": null,
            "overflow_y": null,
            "padding": null,
            "right": null,
            "top": null,
            "visibility": null,
            "width": null
          }
        },
        "89ebf360b3f74a17829e4cf762d4dfbd": {
          "model_module": "@jupyter-widgets/controls",
          "model_name": "DescriptionStyleModel",
          "model_module_version": "1.5.0",
          "state": {
            "_model_module": "@jupyter-widgets/controls",
            "_model_module_version": "1.5.0",
            "_model_name": "DescriptionStyleModel",
            "_view_count": null,
            "_view_module": "@jupyter-widgets/base",
            "_view_module_version": "1.2.0",
            "_view_name": "StyleView",
            "description_width": ""
          }
        }
      }
    },
    "accelerator": "GPU",
    "gpuClass": "premium"
  },
  "nbformat": 4,
  "nbformat_minor": 0
}