{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "import os\n",
    "from sklearn.model_selection import train_test_split\n",
    "from datasets import load_dataset\n",
    "from artifact_classification.utils import ConfigLoader"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Updating with:\n",
      "{'config': 'testing', 'fast_dev_run': True, 'dataset': 'james-burton/BritishMuseum', 'wandb_proj_name': 'British Museum', 'model_base': 'google/efficientnet-b3', 'problem_type': 'image', 'lower_lim': 5, 'label_col': 'Object type'}\n",
      "\n",
      "\n",
      "{'config': 'testing', 'fast_dev_run': True, 'do_train': True, 'do_predict': True, 'batch_size': 128, 'output_root': 'models/', 'num_epochs': 100, 'early_stopping_patience': 5, 'grad_accumulation_steps': 1, 'seed': 42, 'logging_steps': 10, 'lr_scheduler': 'linear', 'warmup_ratio': 0, 'weight_decay': 0, 'device': 'cuda', 'num_workers': 1, 'resume_from_checkpoint': False, 'predict_batch_size': 16, 'save_total_limit': 1, 'lr': 5e-05, 'pytorch2_0': True, 'max_length': 512, 'text_column': 'Description', 'fp16': True, 'testset_size': 0.1, 'dataset': 'james-burton/BritishMuseum', 'wandb_proj_name': 'British Museum', 'model_base': 'google/efficientnet-b3', 'problem_type': 'image', 'lower_lim': 5, 'label_col': 'Object type'}\n",
      "\n"
     ]
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "cca9a5e0c5f2487ea3ad65c183da9c90",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Resolving data files:   0%|          | 0/22 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "431f0e5b4ec84c8693b5c8c18525f810",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Resolving data files:   0%|          | 0/22 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    },
    {
     "data": {
      "application/vnd.jupyter.widget-view+json": {
       "model_id": "afda95f374c1487584af43d91ba321df",
       "version_major": 2,
       "version_minor": 0
      },
      "text/plain": [
       "Loading dataset shards:   0%|          | 0/21 [00:00<?, ?it/s]"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "config = \"testing\"\n",
    "args = ConfigLoader(config, \"../configs/train_bm_configs.yaml\", \"../configs/train_bm_default.yaml\")\n",
    "\n",
    "############################## Load dataset ##############################\n",
    "# Load dataset, filter out na inputs and labels and encode labels (as label column can change)\n",
    "\n",
    "\n",
    "label_cols = [\"Object type\", \"Culture\", \"Materials\", \"Production place\"]\n",
    "split_sizes = [0.1, 0.12, 0.13, 0.15, 0.2]\n",
    "\n",
    "ds_lim = load_dataset(args.dataset)[\"train\"].to_pandas()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Split size 0.1 failed\n",
      "Split size 0.12 failed\n",
      "Label col Object type Split size 0.13 passed\n",
      "Label col Object type Split size 0.15 passed\n",
      "Label col Object type Split size 0.2 passed\n",
      "Split size 0.1 failed\n",
      "Split size 0.12 failed\n",
      "Label col Culture Split size 0.13 passed\n",
      "Label col Culture Split size 0.15 passed\n",
      "Label col Culture Split size 0.2 passed\n",
      "Split size 0.1 failed\n",
      "Split size 0.12 failed\n",
      "Label col Materials Split size 0.13 passed\n",
      "Label col Materials Split size 0.15 passed\n",
      "Label col Materials Split size 0.2 passed\n",
      "Split size 0.1 failed\n",
      "Split size 0.12 failed\n",
      "Label col Production place Split size 0.13 passed\n",
      "Label col Production place Split size 0.15 passed\n",
      "Label col Production place Split size 0.2 passed\n"
     ]
    }
   ],
   "source": [
    "def test_split_size(split_size, label_col, ds_lim):\n",
    "    try:\n",
    "        ds_lim.dropna(subset=[label_col])\n",
    "        num_counts = ds_lim[label_col].value_counts()\n",
    "        ds_lim = ds_lim[ds_lim[label_col].isin(num_counts[num_counts > args.lower_lim].index)]\n",
    "\n",
    "        train, val_test = train_test_split(\n",
    "            ds_lim,\n",
    "            stratify=ds_lim[label_col],\n",
    "            test_size=2 * split_size,\n",
    "            random_state=42,\n",
    "        )\n",
    "        val, test = train_test_split(\n",
    "            val_test, stratify=val_test[label_col], test_size=0.5, random_state=42\n",
    "        )\n",
    "        print(f\"Label col {label_col} Split size {split_size} passed\")\n",
    "    except ValueError:\n",
    "        print(f\"Split size {split_size} failed\")\n",
    "\n",
    "\n",
    "for label_col in label_cols:\n",
    "    for split_size in split_sizes:\n",
    "        test_split_size(split_size, label_col, ds_lim)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import yaml\n",
    "\n",
    "with open(\"../configs/train_configs.yaml\", \"r\") as file:\n",
    "    configs = list(yaml.safe_load_all(file))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'om3-white_material_bm-pretrn om3-white_name_bm-pretrn om3-3Dwhite_material_bm-pretrn om3-3Dwhite_name_bm-pretrn om3-3Dwhite-1frame_material_bm-pretrn om3-3Dwhite-1frame_name_bm-pretrn om4-white_material_bm-pretrn om4-white_name_bm-pretrn om4-3Dwhite_material_bm-pretrn om4-3Dwhite_name_bm-pretrn om4-3Dwhite-1frame_material_bm-pretrn om4-3Dwhite-1frame_name_bm-pretrn om5-white_material_bm-pretrn om5-white_name_bm-pretrn om5-3Dwhite_material_bm-pretrn om5-3Dwhite_name_bm-pretrn om5-3Dwhite-1frame_material_bm-pretrn om5-3Dwhite-1frame_name_bm-pretrn om6-white_material_bm-pretrn om6-white_name_bm-pretrn om6-3Dwhite_material_bm-pretrn om6-3Dwhite_name_bm-pretrn om6-3Dwhite-1frame_material_bm-pretrn om6-3Dwhite-1frame_name_bm-pretrn'"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "\" \".join(\n",
    "    [cfg[\"config\"] for cfg in configs if \"bm\" in cfg[\"config\"] and \"num\" not in cfg[\"config\"]]\n",
    ")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[{'config': 'testing',\n",
       "  'fast_dev_run': True,\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm3_type',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Object type',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Materials',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Culture',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3txt_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Materials',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3txt_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Culture',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-white_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Object type',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-white_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Materials',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-white_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Culture',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-3Dwhite_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Object type',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-3Dwhite_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Materials',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-3Dwhite_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Culture',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-3Dwhite-1frame_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Object type',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-3Dwhite-1frame_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Materials',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm3-3Dwhite-1frame_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 3,\n",
       "  'label_col': 'Culture',\n",
       "  'testset_size': 0.205},\n",
       " {'config': 'bm4_type',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm4_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm4_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm4txt_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm4txt_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm4-white_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm4-white_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm4-white_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm4-3Dwhite_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm4-3Dwhite_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm4-3Dwhite_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm4-3Dwhite-1frame_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm4-3Dwhite-1frame_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm4-3Dwhite-1frame_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 4,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm5_type',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm5_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm5_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm5txt_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm5txt_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm5-white_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm5-white_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm5-white_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm5-3Dwhite_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm5-3Dwhite_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm5-3Dwhite_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm5-3Dwhite-1frame_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm5-3Dwhite-1frame_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm5-3Dwhite-1frame_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 5,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm6_type',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm6_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm6_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm6txt_material',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm6txt_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'microsoft/deberta-v3-base',\n",
       "  'problem_type': 'text',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm6-white_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm6-white_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm6-white_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-white',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm6-3Dwhite_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm6-3Dwhite_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm6-3Dwhite_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Culture'},\n",
       " {'config': 'bm6-3Dwhite-1frame_type',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Object type'},\n",
       " {'config': 'bm6-3Dwhite-1frame_material',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Materials'},\n",
       " {'config': 'bm6-3Dwhite-1frame_culture',\n",
       "  'dataset': 'james-burton/BritishMuseum-3Dwhite-1frame',\n",
       "  'wandb_proj_name': 'British Museum',\n",
       "  'model_base': 'google/efficientnet-b3',\n",
       "  'problem_type': 'image',\n",
       "  'lower_lim': 6,\n",
       "  'label_col': 'Culture'}]"
      ]
     },
     "execution_count": 18,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "configs"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "ArtifactClassification",
   "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.10.12"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}