Resnet.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 9,
   "id": "3babdee1-33d9-4458-b230-ac9e32c904b5",
   "metadata": {},
   "outputs": [],
   "source": [
    "import numpy as np\n",
    "import os\n",
    "from sklearn.metrics import accuracy_score, precision_score, f1_score, roc_curve, roc_auc_score\n",
    "from skimage.io import imread\n",
    "from skimage.transform import resize\n",
    "import matplotlib.pyplot as plt\n",
    "from sklearn.model_selection import train_test_split\n",
    "from tensorflow.keras.applications import ResNet152V2\n",
    "from tensorflow.keras.layers import Dense, GlobalAveragePooling2D\n",
    "from tensorflow.keras.models import Model\n",
    "from tensorflow.keras.optimizers import Adam\n",
    "from sklearn.preprocessing import LabelEncoder\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "ee990360-967f-4687-9770-88d607dd760f",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "# Function to load images and labels from folder\n",
    "def load_images_and_labels(folder, target_size=(224, 224)):\n",
    "    images = []\n",
    "    labels = []\n",
    "    for subfolder in os.listdir(folder):\n",
    "        subfolder_path = os.path.join(folder, subfolder)\n",
    "        label = subfolder\n",
    "        for filename in os.listdir(subfolder_path):\n",
    "            image_path = os.path.join(subfolder_path, filename)\n",
    "            try:\n",
    "                img = imread(image_path)\n",
    "                if img is not None:\n",
    "                    img_resized = resize(img, target_size, anti_aliasing=True)  # Resize image to target size\n",
    "                    images.append(img_resized)  \n",
    "                    labels.append(label)\n",
    "                else:\n",
    "                    print(f\"Error reading image: {image_path}\")\n",
    "            except Exception as e:\n",
    "                print(f\"Error loading image: {image_path} - {str(e)}\")\n",
    "    return np.array(images), np.array(labels)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "b9b382e8-d88f-4beb-b0ca-419b61f66f43",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Folder paths\n",
    "train_folder = 'C:/Users/KIIT/Downloads/Cashew/Cashew/train'\n",
    "test_folder ='C:/Users/KIIT/Downloads/Cashew/Cashew/test'\n",
    "val_folder ='C:/Users/KIIT/Downloads/Cashew/Cashew/val'"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "14c69ac0-af9d-4614-80db-cf4f5bdd59df",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust475_.jpg - Unable to allocate 1.15 MiB for an array with shape (224, 224, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust479_.jpg - Unable to allocate 1.15 MiB for an array with shape (224, 224, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust47_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust480_.jpg - Unable to allocate 1.15 MiB for an array with shape (224, 224, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust482_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust483_.jpg - Unable to allocate 1.15 MiB for an array with shape (224, 224, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust484_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust485_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust486_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust488_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust489_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust48_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust490_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust491_.jpg - Unable to allocate 1.15 MiB for an array with shape (224, 224, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust492_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust493_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust494_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust495_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust496_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust497_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust498_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust499_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust49_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust4_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust501_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust502_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust503_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust504_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust505_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust507_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust508_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust509_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust50_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust510_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust512_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust515_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust516_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust518_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust519_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust520_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust521_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust522_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust523_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust524_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust525_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust526_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust527_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust529_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust52_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust530_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust532_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust534_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust535_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust536_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust537_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust539_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust53_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust540_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust541_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust543_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust544_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust545_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust548_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust549_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust550_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust551_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust553_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust554_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust557_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust558_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust55_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust560_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust562_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust563_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust564_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust565_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust566_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust567_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust569_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust56_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust571_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust572_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust573_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust574_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust581_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust582_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust583_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust584_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust585_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust586_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust587_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust588_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust589_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust58_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust590_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust591_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust592_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust594_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust596_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust597_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust598_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust599_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust59_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust5_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust600_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust601_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust602_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust604_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust605_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust608_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust609_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust60_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust611_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust612_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust613_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust615_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust616_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust618_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust619_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust622_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust624_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust625_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust626_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust627_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust629_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust62_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust632_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust633_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust634_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust636_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust637_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust638_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust639_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust63_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust641_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust643_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust644_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust645_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust646_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust647_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust648_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust649_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust64_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust651_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust652_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust653_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust654_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust655_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust656_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust657_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust658_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust659_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust65_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust660_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust661_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust663_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust664_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust666_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust667_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust669_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust66_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust670_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust671_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust675_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust676_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust677_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust678_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust67_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust681_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust682_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust685_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust686_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust687_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust688_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust689_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust68_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust690_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust691_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust692_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust694_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust695_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust697_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust698_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust6_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust701_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust702_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust705_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust706_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust707_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust708_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust709_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust70_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust710_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust711_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust712_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust714_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust715_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust716_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust718_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust719_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust71_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust720_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust721_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust722_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust723_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust724_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust725_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust726_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust727_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust728_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust729_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust72_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust730_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust731_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust732_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust733_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust734_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust736_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust737_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust738_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust739_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust73_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust740_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust742_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust743_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust744_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust745_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust746_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust747_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust748_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust74_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust751_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust752_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust754_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust755_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust756_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust757_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust758_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust759_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust75_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust761_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust762_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust763_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust764_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust765_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust766_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust767_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust769_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust76_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust770_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust771_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust775_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust777_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust778_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust779_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust780_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust781_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust783_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust785_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust786_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust787_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust788_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust789_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust78_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust790_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust791_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust792_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust793_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust794_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust795_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust796_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust797_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust798_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust799_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust79_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust7_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust800_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust801_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust802_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust804_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust806_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust807_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust809_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust810_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust811_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust812_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust814_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust815_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust816_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust818_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust820_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust821_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust822_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust823_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust825_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust826_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust827_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust828_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust829_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust82_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust830_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust832_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust833_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust834_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust835_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust838_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust839_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust83_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust841_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust842_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust843_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust845_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust846_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust847_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust849_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust84_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust852_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust854_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust855_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust856_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust857_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust858_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust859_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust85_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust860_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust861_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust864_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust865_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust866_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust867_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust868_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust86_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust870_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust871_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust872_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust873_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust874_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust875_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust878_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust879_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust87_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust881_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust882_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust884_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust885_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust887_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust888_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust889_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust890_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust892_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust893_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust894_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust895_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust896_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust897_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust898_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust899_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust89_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust8_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust900_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust902_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust905_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust907_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust909_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust90_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust910_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust911_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust912_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust913_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust914_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust916_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust919_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust91_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust920_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust921_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust922_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust923_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust925_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust926_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust927_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust928_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust92_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust930_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust931_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust932_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust933_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust934_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust935_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust936_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust937_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust938_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust939_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust941_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust942_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust943_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust944_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust945_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust946_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust947_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust948_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust94_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust950_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust951_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust953_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust955_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n",
      "Error loading image: C:/Users/KIIT/Downloads/Cashew/Cashew/train\\red rust\\red rust956_.jpg - Unable to allocate 3.66 MiB for an array with shape (400, 400, 3) and data type float64\n"
     ]
    },
    {
     "ename": "MemoryError",
     "evalue": "Unable to allocate 6.72 GiB for an array with shape (5996, 224, 224, 3) and data type float64",
     "output_type": "error",
     "traceback": [
      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[1;31mMemoryError\u001b[0m                               Traceback (most recent call last)",
      "Cell \u001b[1;32mIn[12], line 2\u001b[0m\n\u001b[0;32m      1\u001b[0m \u001b[38;5;66;03m# Load images and labels for training, testing, and validation\u001b[39;00m\n\u001b[1;32m----> 2\u001b[0m X_train, y_train \u001b[38;5;241m=\u001b[39m load_images_and_labels(train_folder)\n\u001b[0;32m      3\u001b[0m X_test, y_test \u001b[38;5;241m=\u001b[39m load_images_and_labels(test_folder)\n\u001b[0;32m      4\u001b[0m X_val, y_val \u001b[38;5;241m=\u001b[39m load_images_and_labels(val_folder)\n",
      "Cell \u001b[1;32mIn[10], line 20\u001b[0m, in \u001b[0;36mload_images_and_labels\u001b[1;34m(folder, target_size)\u001b[0m\n\u001b[0;32m     18\u001b[0m         \u001b[38;5;28;01mexcept\u001b[39;00m \u001b[38;5;167;01mException\u001b[39;00m \u001b[38;5;28;01mas\u001b[39;00m e:\n\u001b[0;32m     19\u001b[0m             \u001b[38;5;28mprint\u001b[39m(\u001b[38;5;124mf\u001b[39m\u001b[38;5;124m\"\u001b[39m\u001b[38;5;124mError loading image: \u001b[39m\u001b[38;5;132;01m{\u001b[39;00mimage_path\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m - \u001b[39m\u001b[38;5;132;01m{\u001b[39;00m\u001b[38;5;28mstr\u001b[39m(e)\u001b[38;5;132;01m}\u001b[39;00m\u001b[38;5;124m\"\u001b[39m)\n\u001b[1;32m---> 20\u001b[0m \u001b[38;5;28;01mreturn\u001b[39;00m np\u001b[38;5;241m.\u001b[39marray(images), np\u001b[38;5;241m.\u001b[39marray(labels)\n",
      "\u001b[1;31mMemoryError\u001b[0m: Unable to allocate 6.72 GiB for an array with shape (5996, 224, 224, 3) and data type float64"
     ]
    }
   ],
   "source": [
    "\n",
    "# Load images and labels for training, testing, and validation\n",
    "X_train, y_train = load_images_and_labels(train_folder)\n",
    "X_test, y_test = load_images_and_labels(test_folder)\n",
    "X_val, y_val = load_images_and_labels(val_folder)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "5eb4804a-65f4-466c-8cc6-15995847a86e",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Convert string labels to integer labels\n",
    "label_encoder = LabelEncoder()\n",
    "y_train = label_encoder.fit_transform(y_train)\n",
    "y_test = label_encoder.transform(y_test)\n",
    "y_val = label_encoder.transform(y_val)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e65997a5-db91-46ee-a9fe-82614cedfbce",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Load pre-trained ResNet152V2 model\n",
    "base_model = ResNet152V2(weights='imagenet', include_top=False)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "b8290da0-0222-4907-92a6-6804413d655c",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Add custom classifier head\n",
    "x = base_model.output\n",
    "x = GlobalAveragePooling2D()(x)\n",
    "x = Dense(1024, activation='relu')(x)\n",
    "predictions = Dense(len(os.listdir(train_folder)), activation='softmax')(x)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fb377b93-25bf-4b19-a0e6-d7e91cb862aa",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Combine base model and custom head\n",
    "model = Model(inputs=base_model.input, outputs=predictions)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "2d323f75-7904-464d-a591-0540d8d3551c",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Freeze layers in base model\n",
    "for layer in base_model.layers:\n",
    "    layer.trainable = False\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "559db750-414b-46c6-8a29-9fa6c2595631",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Compile the model\n",
    "model.compile(optimizer=Adam(), loss='sparse_categorical_crossentropy', metrics=['accuracy'])\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "fad9f902-fc43-44e2-a187-0db67800e775",
   "metadata": {},
   "outputs": [],
   "source": [
    "# Train the model\n",
    "model.fit(X_train, y_train, epochs=20, batch_size=32, validation_data=(X_val, y_val))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "ef7550a8-a127-4daa-adbf-97160352a682",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.metrics import accuracy_score, precision_score, f1_score, roc_auc_score, recall_score\n",
    "import numpy as np\n",
    "\n",
    "# Predict probability estimates\n",
    "train_pred_proba = model.predict(X_train)\n",
    "val_pred_proba = model.predict(X_val)\n",
    "test_pred_proba = model.predict(X_test)\n",
    "\n",
    "# Get predicted classes\n",
    "train_pred = np.argmax(train_pred_proba, axis=1)\n",
    "val_pred = np.argmax(val_pred_proba, axis=1)\n",
    "test_pred = np.argmax(test_pred_proba, axis=1)\n",
    "\n",
    "# Calculate evaluation metrics\n",
    "train_accuracy = accuracy_score(y_train, train_pred)\n",
    "train_precision = precision_score(y_train, train_pred, average='weighted')\n",
    "train_f1 = f1_score(y_train, train_pred, average='weighted')\n",
    "train_recall = recall_score(y_train, train_pred, average='weighted')\n",
    "train_roc_auc = roc_auc_score(y_train, train_pred_proba, average='macro', multi_class='ovr')\n",
    "\n",
    "val_accuracy = accuracy_score(y_val, val_pred)\n",
    "val_precision = precision_score(y_val, val_pred, average='weighted')\n",
    "val_f1 = f1_score(y_val, val_pred, average='weighted')\n",
    "val_recall = recall_score(y_val, val_pred, average='weighted')\n",
    "val_roc_auc = roc_auc_score(y_val, val_pred_proba, average='macro', multi_class='ovr')\n",
    "\n",
    "test_accuracy = accuracy_score(y_test, test_pred)\n",
    "test_precision = precision_score(y_test, test_pred, average='weighted')\n",
    "test_f1 = f1_score(y_test, test_pred, average='weighted')\n",
    "test_recall = recall_score(y_test, test_pred, average='weighted')\n",
    "test_roc_auc = roc_auc_score(y_test, test_pred_proba, average='macro', multi_class='ovr')\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "8c3b258f-7de4-4e86-906d-07caf1a7a689",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "# Print evaluation metrics\n",
    "print(\"Train data Accuracy:\", train_accuracy)\n",
    "print(\"Train data Precision:\", train_precision)\n",
    "print(\"Train data Recall:\", train_recall)\n",
    "print(\"Train data F1 Score:\", train_f1)\n",
    "print(\"Train data ROC AUC Score:\", train_roc_auc)\n",
    "\n",
    "print(\"\\nValidation data Accuracy:\", val_accuracy)\n",
    "print(\"Validation data Precision:\", val_precision)\n",
    "print(\"Validation data Recall:\", val_recall)\n",
    "print(\"Validation data F1 Score:\", val_f1)\n",
    "print(\"Validation data ROC AUC Score:\", val_roc_auc)\n",
    "\n",
    "print(\"\\nTest data Accuracy:\", test_accuracy)\n",
    "print(\"Test data Precision:\", test_precision)\n",
    "print(\"Test data Recall:\", test_recall)\n",
    "print(\"Test data F1 Score:\", test_f1)\n",
    "print(\"Test data ROC AUC Score:\", test_roc_auc)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e3991ad1-4a4b-4c78-b227-06b391cfce4f",
   "metadata": {},
   "outputs": [],
   "source": [
    "from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score, roc_auc_score\n",
    "\n",
    "def calculate_metrics(y_true, y_pred, y_prob):\n",
    "    accuracy = accuracy_score(y_true, y_pred)\n",
    "    precision = precision_score(y_true, y_pred)\n",
    "    recall = recall_score(y_true, y_pred)\n",
    "    f1 = f1_score(y_true, y_pred)\n",
    "    roc_auc = roc_auc_score(y_true, y_prob)\n",
    "    return accuracy, precision, recall, f1, roc_auc\n",
    "\n",
    "def print_metrics_table(train_metrics, val_metrics, test_metrics):\n",
    "    print(\"Metrics\\t\\t\\tTrain\\t\\tValidation\\tTest\")\n",
    "    print(\"Accuracy\\t\\t{:.4f}\\t\\t{:.4f}\\t\\t{:.4f}\".format(train_metrics[0], val_metrics[0], test_metrics[0]))\n",
    "    print(\"Precision\\t\\t{:.4f}\\t\\t{:.4f}\\t\\t{:.4f}\".format(train_metrics[1], val_metrics[1], test_metrics[1]))\n",
    "    print(\"Recall\\t\\t\\t{:.4f}\\t\\t{:.4f}\\t\\t{:.4f}\".format(train_metrics[2], val_metrics[2], test_metrics[2]))\n",
    "    print(\"F1 Score\\t\\t{:.4f}\\t\\t{:.4f}\\t\\t{:.4f}\".format(train_metrics[3], val_metrics[3], test_metrics[3]))\n",
    "    print(\"ROC AUC Score\\t\\t{:.4f}\\t\\t{:.4f}\\t\\t{:.4f}\".format(train_metrics[4], val_metrics[4], test_metrics[4]))\n",
    "\n",
    "# Assuming train_metrics, val_metrics, and test_metrics are tuples/lists containing metrics in the order: (accuracy, precision, recall, f1_score, roc_auc_score)\n",
    "train_metrics = (train_accuracy, train_precision, train_recall, train_f1, train_roc_auc)\n",
    "val_metrics = (val_accuracy, val_precision, val_recall, val_f1, val_roc_auc)\n",
    "test_metrics = (test_accuracy, test_precision, test_recall, test_f1, test_roc_auc)\n",
    "\n",
    "print_metrics_table(train_metrics, val_metrics, test_metrics)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "4220409d-72a1-4708-8ae7-cfbcfa672fdd",
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.11.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}