speech_pipeline_experiments.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "id": "yqZn5s2Ad8ZX"
   },
   "outputs": [],
   "source": [
    "import os\n",
    "os.chdir('/Users/davidmcdevitt/Documents/GitHub/speech-pipeline')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "m7BGPx3uoCyv",
    "outputId": "9c05ae96-7817-4483-96dc-72acee4db61a"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Already up to date.\r\n"
     ]
    }
   ],
   "source": [
    "!git pull"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "dzxtZG09jQI4",
    "outputId": "fc45d305-47d3-4505-d92e-a398ab63a0fe"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Requirement already satisfied: numpy in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 1)) (1.22.4)\n",
      "Requirement already satisfied: pandas in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 2)) (1.3.4)\n",
      "Requirement already satisfied: torch in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 3)) (1.10.1)\n",
      "Requirement already satisfied: torchaudio in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 4)) (0.10.1)\n",
      "Requirement already satisfied: matplotlib in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 5)) (3.5.1)\n",
      "Requirement already satisfied: IPython in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 7)) (8.14.0)\n",
      "Requirement already satisfied: pydub in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 8)) (0.25.1)\n",
      "Requirement already satisfied: scikit-learn in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from -r requirements.txt (line 9)) (0.24.2)\n",
      "Requirement already satisfied: python-dateutil>=2.7.3 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from pandas->-r requirements.txt (line 2)) (2.8.2)\n",
      "Requirement already satisfied: pytz>=2017.3 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from pandas->-r requirements.txt (line 2)) (2021.3)\n",
      "Requirement already satisfied: typing-extensions in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from torch->-r requirements.txt (line 3)) (4.4.0)\n",
      "Requirement already satisfied: packaging>=20.0 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from matplotlib->-r requirements.txt (line 5)) (23.0)\n",
      "Requirement already satisfied: pyparsing>=2.2.1 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from matplotlib->-r requirements.txt (line 5)) (3.0.4)\n",
      "Requirement already satisfied: pillow>=6.2.0 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from matplotlib->-r requirements.txt (line 5)) (8.4.0)\n",
      "Requirement already satisfied: kiwisolver>=1.0.1 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from matplotlib->-r requirements.txt (line 5)) (1.3.1)\n",
      "Requirement already satisfied: fonttools>=4.22.0 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from matplotlib->-r requirements.txt (line 5)) (4.25.0)\n",
      "Requirement already satisfied: cycler>=0.10 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from matplotlib->-r requirements.txt (line 5)) (0.10.0)\n",
      "Requirement already satisfied: traitlets>=5 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (5.9.0)\n",
      "Requirement already satisfied: backcall in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (0.2.0)\n",
      "Requirement already satisfied: pickleshare in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (0.7.5)\n",
      "Requirement already satisfied: stack-data in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (0.2.0)\n",
      "Requirement already satisfied: pygments>=2.4.0 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (2.16.1)\n",
      "Requirement already satisfied: matplotlib-inline in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (0.1.2)\n",
      "Requirement already satisfied: appnope in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (0.1.2)\n",
      "Requirement already satisfied: decorator in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (4.4.2)\n",
      "Requirement already satisfied: prompt-toolkit!=3.0.37,<3.1.0,>=3.0.30 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (3.0.39)\n",
      "Requirement already satisfied: jedi>=0.16 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (0.18.0)\n",
      "Requirement already satisfied: pexpect>4.3 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from IPython->-r requirements.txt (line 7)) (4.8.0)\n",
      "Requirement already satisfied: scipy>=0.19.1 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from scikit-learn->-r requirements.txt (line 9)) (1.7.1)\n",
      "Requirement already satisfied: threadpoolctl>=2.0.0 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from scikit-learn->-r requirements.txt (line 9)) (2.2.0)\n",
      "Requirement already satisfied: joblib>=0.11 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from scikit-learn->-r requirements.txt (line 9)) (1.1.0)\n",
      "Requirement already satisfied: six in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from cycler>=0.10->matplotlib->-r requirements.txt (line 5)) (1.16.0)\n",
      "Requirement already satisfied: parso<0.9.0,>=0.8.0 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from jedi>=0.16->IPython->-r requirements.txt (line 7)) (0.8.2)\n",
      "Requirement already satisfied: ptyprocess>=0.5 in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from pexpect>4.3->IPython->-r requirements.txt (line 7)) (0.7.0)\n",
      "Requirement already satisfied: wcwidth in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from prompt-toolkit!=3.0.37,<3.1.0,>=3.0.30->IPython->-r requirements.txt (line 7)) (0.2.5)\n",
      "Requirement already satisfied: executing in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from stack-data->IPython->-r requirements.txt (line 7)) (0.8.3)\n",
      "Requirement already satisfied: pure-eval in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from stack-data->IPython->-r requirements.txt (line 7)) (0.2.2)\n",
      "Requirement already satisfied: asttokens in /Users/davidmcdevitt/opt/anaconda3/lib/python3.9/site-packages (from stack-data->IPython->-r requirements.txt (line 7)) (2.0.5)\n"
     ]
    }
   ],
   "source": [
    "!pip install -r requirements.txt"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "4aTcwOR62gN9"
   },
   "source": [
    "# Build Datasets"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "id": "So_YkL2r2DgK"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "mkdir: ./experiments/: File exists\r\n"
     ]
    }
   ],
   "source": [
    "!mkdir ./experiments/\n",
    "!mkdir ./experiments/seed_3class_transition\n",
    "!mkdir ./experiments/seed_3class_transition/data\n",
    "!mkdir ./experiments/seed_3class_transition/data/children\n",
    "!mkdir ./experiments/seed_3class_transition/data/adults\n",
    "!mkdir ./experiments/seed_3class_transition/data/mixed\n",
    "!mkdir ./experiments/seed_3class_transition/data/transitions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "4sW2fP652dk_",
    "outputId": "72b9a877-f1b4-4ccd-8605-ccfb0871d340"
   },
   "outputs": [],
   "source": [
    "!unzip -q /Users/davidmcdevitt/Downloads/seed_children.zip -d ./experiments/seed_3class_transition/data\n",
    "!unzip -q /Users/davidmcdevitt/Downloads/seed_adults.zip -d ./experiments/seed_3class_transition/data\n",
    "#!unzip -q /Users/davidmcdevitt/Downloads/darcy_children.zip -d ./experiments/seed_2class/data\n",
    "#!unzip -q /Users/davidmcdevitt/Downloads/cv_adults.zip -d ./experiments/seed_2class/data"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "T1EegnSXw0iw"
   },
   "source": [
    "Full 2 Class"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "colab": {
     "base_uri": "https://localhost:8080/"
    },
    "id": "oDFsrboXlOam",
    "outputId": "e12a59db-4591-452d-a5fe-01b5f4294fcd"
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Building datasets\n",
      "Child Obs: 11977\n",
      "tensor([0.3153, 0.3326, 0.3521])\n",
      "Epoch: 0, Loss: 0.97, Accuracy: 0.56, Time: 878.11 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "[W NNPACK.cpp:79] Could not initialize NNPACK! Reason: Unsupported hardware.\n",
      "Accuracy: 0.61, Total items: 3233\n",
      "Weighted F1 Score: 0.61\n",
      "Confusion Matrix:\n",
      "[[637 239 229]\n",
      " [221 678 224]\n",
      " [155 187 663]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 1, Loss: 0.90, Accuracy: 0.64, Time: 1617.35 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.65, Total items: 3233\n",
      "Weighted F1 Score: 0.65\n",
      "Confusion Matrix:\n",
      "[[722 216 167]\n",
      " [251 733 139]\n",
      " [167 190 648]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 2, Loss: 0.87, Accuracy: 0.67, Time: 786.96 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.67, Total items: 3233\n",
      "Weighted F1 Score: 0.67\n",
      "Confusion Matrix:\n",
      "[[788 168 149]\n",
      " [244 723 156]\n",
      " [159 175 671]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 3, Loss: 0.86, Accuracy: 0.68, Time: 827.92 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.68, Total items: 3233\n",
      "Weighted F1 Score: 0.68\n",
      "Confusion Matrix:\n",
      "[[795 161 149]\n",
      " [283 694 146]\n",
      " [140 160 705]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 4, Loss: 0.84, Accuracy: 0.70, Time: 844.31 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.70, Total items: 3233\n",
      "Weighted F1 Score: 0.70\n",
      "Confusion Matrix:\n",
      "[[805 182 118]\n",
      " [233 768 122]\n",
      " [150 167 688]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 5, Loss: 0.83, Accuracy: 0.71, Time: 829.35 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.70, Total items: 3233\n",
      "Weighted F1 Score: 0.70\n",
      "Confusion Matrix:\n",
      "[[768 177 160]\n",
      " [197 763 163]\n",
      " [123 136 746]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 6, Loss: 0.82, Accuracy: 0.72, Time: 5175.11 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.70, Total items: 3233\n",
      "Weighted F1 Score: 0.70\n",
      "Confusion Matrix:\n",
      "[[809 168 128]\n",
      " [225 781 117]\n",
      " [155 172 678]]\n",
      "Break Count: 1\n",
      "Epoch: 7, Loss: 0.82, Accuracy: 0.72, Time: 704.78 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.72, Total items: 3233\n",
      "Weighted F1 Score: 0.72\n",
      "Confusion Matrix:\n",
      "[[791 173 141]\n",
      " [218 810  95]\n",
      " [112 177 716]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 8, Loss: 0.81, Accuracy: 0.73, Time: 755.04 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.73, Total items: 3233\n",
      "Weighted F1 Score: 0.73\n",
      "Confusion Matrix:\n",
      "[[854 151 100]\n",
      " [200 828  95]\n",
      " [133 182 690]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 9, Loss: 0.81, Accuracy: 0.73, Time: 788.92 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.72, Total items: 3233\n",
      "Weighted F1 Score: 0.72\n",
      "Confusion Matrix:\n",
      "[[774 194 137]\n",
      " [195 790 138]\n",
      " [128 112 765]]\n",
      "Break Count: 1\n",
      "Epoch: 10, Loss: 0.80, Accuracy: 0.74, Time: 799.94 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.73, Total items: 3233\n",
      "Weighted F1 Score: 0.73\n",
      "Confusion Matrix:\n",
      "[[857 152  96]\n",
      " [227 785 111]\n",
      " [148 152 705]]\n",
      "Break Count: 2\n",
      "Epoch: 11, Loss: 0.80, Accuracy: 0.74, Time: 820.47 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.72, Total items: 3233\n",
      "Weighted F1 Score: 0.72\n",
      "Confusion Matrix:\n",
      "[[805 154 146]\n",
      " [200 805 118]\n",
      " [126 148 731]]\n",
      "Break Count: 3\n",
      "Epoch: 12, Loss: 0.80, Accuracy: 0.75, Time: 5295.85 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.74, Total items: 3233\n",
      "Weighted F1 Score: 0.74\n",
      "Confusion Matrix:\n",
      "[[858 138 109]\n",
      " [215 802 106]\n",
      " [124 150 731]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 13, Loss: 0.80, Accuracy: 0.74, Time: 704.23 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.74, Total items: 3233\n",
      "Weighted F1 Score: 0.74\n",
      "Confusion Matrix:\n",
      "[[893 140  72]\n",
      " [218 792 113]\n",
      " [176 118 711]]\n",
      "Break Count: 1\n",
      "Epoch: 14, Loss: 0.79, Accuracy: 0.75, Time: 747.29 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.74, Total items: 3233\n",
      "Weighted F1 Score: 0.74\n",
      "Confusion Matrix:\n",
      "[[833 149 123]\n",
      " [184 843  96]\n",
      " [125 177 703]]\n",
      "Break Count: 2\n",
      "Epoch: 15, Loss: 0.79, Accuracy: 0.75, Time: 808.39 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.74, Total items: 3233\n",
      "Weighted F1 Score: 0.74\n",
      "Confusion Matrix:\n",
      "[[896 117  92]\n",
      " [233 773 117]\n",
      " [154 116 735]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 16, Loss: 0.79, Accuracy: 0.76, Time: 821.77 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.75, Total items: 3233\n",
      "Weighted F1 Score: 0.75\n",
      "Confusion Matrix:\n",
      "[[853 139 113]\n",
      " [212 814  97]\n",
      " [111 125 769]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 17, Loss: 0.78, Accuracy: 0.76, Time: 838.01 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.75, Total items: 3233\n",
      "Weighted F1 Score: 0.75\n",
      "Confusion Matrix:\n",
      "[[883 149  73]\n",
      " [222 821  80]\n",
      " [145 143 717]]\n",
      "Break Count: 1\n",
      "Epoch: 18, Loss: 0.78, Accuracy: 0.76, Time: 854.73 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.75, Total items: 3233\n",
      "Weighted F1 Score: 0.75\n",
      "Confusion Matrix:\n",
      "[[831 201  73]\n",
      " [163 882  78]\n",
      " [139 144 722]]\n",
      "Break Count: 2\n",
      "Epoch: 19, Loss: 0.79, Accuracy: 0.75, Time: 864.03 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.75, Total items: 3233\n",
      "Weighted F1 Score: 0.75\n",
      "Confusion Matrix:\n",
      "[[869 123 113]\n",
      " [211 792 120]\n",
      " [158  96 751]]\n",
      "Break Count: 3\n",
      "Epoch: 20, Loss: 0.78, Accuracy: 0.77, Time: 853.78 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.77, Total items: 3233\n",
      "Weighted F1 Score: 0.77\n",
      "Confusion Matrix:\n",
      "[[895 124  86]\n",
      " [176 849  98]\n",
      " [123 128 754]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 21, Loss: 0.78, Accuracy: 0.77, Time: 860.55 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.75, Total items: 3233\n",
      "Weighted F1 Score: 0.75\n",
      "Confusion Matrix:\n",
      "[[836 151 118]\n",
      " [210 828  85]\n",
      " [109 143 753]]\n",
      "Break Count: 1\n",
      "Epoch: 22, Loss: 0.78, Accuracy: 0.77, Time: 2027.81 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.76, Total items: 3233\n",
      "Weighted F1 Score: 0.76\n",
      "Confusion Matrix:\n",
      "[[908 105  92]\n",
      " [230 754 139]\n",
      " [108  96 801]]\n",
      "Break Count: 2\n",
      "Epoch: 23, Loss: 0.78, Accuracy: 0.76, Time: 752.63 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[894 112  99]\n",
      " [210 831  82]\n",
      " [118 106 781]]\n",
      "Break Count: 3\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Epoch: 24, Loss: 0.78, Accuracy: 0.77, Time: 790.71 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.76, Total items: 3233\n",
      "Weighted F1 Score: 0.76\n",
      "Confusion Matrix:\n",
      "[[881 111 113]\n",
      " [235 795  93]\n",
      " [102 135 768]]\n",
      "Break Count: 4\n",
      "Epoch: 25, Loss: 0.78, Accuracy: 0.77, Time: 809.69 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.77, Total items: 3233\n",
      "Weighted F1 Score: 0.77\n",
      "Confusion Matrix:\n",
      "[[869 146  90]\n",
      " [178 857  88]\n",
      " [100 140 765]]\n",
      "Break Count: 5\n",
      "Epoch: 26, Loss: 0.77, Accuracy: 0.77, Time: 817.59 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.76, Total items: 3233\n",
      "Weighted F1 Score: 0.76\n",
      "Confusion Matrix:\n",
      "[[799 200 106]\n",
      " [143 847 133]\n",
      " [101  90 814]]\n",
      "Break Count: 6\n",
      "Epoch: 27, Loss: 0.77, Accuracy: 0.77, Time: 816.87 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[871 122 112]\n",
      " [171 872  80]\n",
      " [ 96 139 770]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 28, Loss: 0.77, Accuracy: 0.77, Time: 838.13 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[908 121  76]\n",
      " [189 841  93]\n",
      " [112 133 760]]\n",
      "Break Count: 1\n",
      "Epoch: 29, Loss: 0.77, Accuracy: 0.78, Time: 848.00 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.77, Total items: 3233\n",
      "Weighted F1 Score: 0.77\n",
      "Confusion Matrix:\n",
      "[[920 120  65]\n",
      " [209 839  75]\n",
      " [146 124 735]]\n",
      "Break Count: 2\n",
      "Epoch: 30, Loss: 0.77, Accuracy: 0.78, Time: 906.53 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[920 110  75]\n",
      " [203 829  91]\n",
      " [122 120 763]]\n",
      "Break Count: 3\n",
      "Epoch: 31, Loss: 0.77, Accuracy: 0.78, Time: 924.47 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.77, Total items: 3233\n",
      "Weighted F1 Score: 0.77\n",
      "Confusion Matrix:\n",
      "[[848 136 121]\n",
      " [157 878  88]\n",
      " [ 89 137 779]]\n",
      "Break Count: 4\n",
      "Epoch: 32, Loss: 0.76, Accuracy: 0.78, Time: 914.42 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.77, Total items: 3233\n",
      "Weighted F1 Score: 0.77\n",
      "Confusion Matrix:\n",
      "[[855 146 104]\n",
      " [164 885  74]\n",
      " [114 149 742]]\n",
      "Break Count: 5\n",
      "Epoch: 33, Loss: 0.77, Accuracy: 0.78, Time: 2409.80 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[867 170  68]\n",
      " [142 868 113]\n",
      " [121 107 777]]\n",
      "Break Count: 6\n",
      "Epoch: 34, Loss: 0.76, Accuracy: 0.78, Time: 1357.05 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[895 114  96]\n",
      " [164 849 110]\n",
      " [102  87 816]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 35, Loss: 0.76, Accuracy: 0.78, Time: 6513.89 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[904 137  64]\n",
      " [193 847  83]\n",
      " [110 137 758]]\n",
      "Break Count: 1\n",
      "Epoch: 36, Loss: 0.76, Accuracy: 0.78, Time: 914.11 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[896 111  98]\n",
      " [203 845  75]\n",
      " [112 101 792]]\n",
      "Break Count: 2\n",
      "Epoch: 37, Loss: 0.76, Accuracy: 0.78, Time: 892.20 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.77, Total items: 3233\n",
      "Weighted F1 Score: 0.77\n",
      "Confusion Matrix:\n",
      "[[839 169  97]\n",
      " [174 871  78]\n",
      " [ 83 145 777]]\n",
      "Break Count: 3\n",
      "Epoch: 38, Loss: 0.76, Accuracy: 0.78, Time: 2595.76 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[925 118  62]\n",
      " [204 840  79]\n",
      " [143  96 766]]\n",
      "Break Count: 4\n",
      "Epoch: 39, Loss: 0.76, Accuracy: 0.78, Time: 880.45 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[823 184  98]\n",
      " [113 953  57]\n",
      " [ 95 141 769]]\n",
      "Break Count: 5\n",
      "Epoch: 40, Loss: 0.76, Accuracy: 0.78, Time: 999.57 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[899 121  85]\n",
      " [186 860  77]\n",
      " [101 120 784]]\n",
      "Break Count: 6\n",
      "Epoch: 41, Loss: 0.76, Accuracy: 0.79, Time: 10641.43 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[900 117  88]\n",
      " [193 824 106]\n",
      " [108  98 799]]\n",
      "Break Count: 7\n",
      "Epoch: 42, Loss: 0.76, Accuracy: 0.79, Time: 703.10 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.77, Total items: 3233\n",
      "Weighted F1 Score: 0.77\n",
      "Confusion Matrix:\n",
      "[[887 136  82]\n",
      " [184 862  77]\n",
      " [123 130 752]]\n",
      "Break Count: 8\n",
      "Epoch: 43, Loss: 0.76, Accuracy: 0.79, Time: 680.67 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[809 178 118]\n",
      " [143 913  67]\n",
      " [ 78 131 796]]\n",
      "Break Count: 9\n",
      "Epoch: 44, Loss: 0.76, Accuracy: 0.78, Time: 1552.75 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[940  98  67]\n",
      " [216 817  90]\n",
      " [131 103 771]]\n",
      "Break Count: 10\n",
      "Epoch: 45, Loss: 0.75, Accuracy: 0.79, Time: 744.87 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[851 175  79]\n",
      " [159 881  83]\n",
      " [102 118 785]]\n",
      "Break Count: 11\n",
      "Epoch: 46, Loss: 0.76, Accuracy: 0.79, Time: 691.13 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[930 102  73]\n",
      " [197 854  72]\n",
      " [133 114 758]]\n",
      "Break Count: 12\n",
      "Epoch: 47, Loss: 0.75, Accuracy: 0.79, Time: 687.21 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[907 105  93]\n",
      " [193 862  68]\n",
      " [109  92 804]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 48, Loss: 0.75, Accuracy: 0.79, Time: 1589.92 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[914 117  74]\n",
      " [213 850  60]\n",
      " [106 112 787]]\n",
      "Break Count: 1\n",
      "Epoch: 49, Loss: 0.75, Accuracy: 0.79, Time: 682.50 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[954  82  69]\n",
      " [201 859  63]\n",
      " [120 115 770]]\n",
      "Break Count: 2\n",
      "Epoch: 50, Loss: 0.75, Accuracy: 0.79, Time: 2773.94 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[888 131  86]\n",
      " [172 874  77]\n",
      " [106 114 785]]\n",
      "Break Count: 3\n",
      "Epoch: 51, Loss: 0.75, Accuracy: 0.79, Time: 6011.56 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[842 174  89]\n",
      " [144 919  60]\n",
      " [100 125 780]]\n",
      "Break Count: 4\n",
      "Epoch: 52, Loss: 0.75, Accuracy: 0.80, Time: 3332.66 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[917 121  67]\n",
      " [170 877  76]\n",
      " [123 107 775]]\n",
      "Break Count: 5\n",
      "Epoch: 53, Loss: 0.75, Accuracy: 0.79, Time: 957.06 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[832 178  95]\n",
      " [102 943  78]\n",
      " [ 82 146 777]]\n",
      "Break Count: 6\n",
      "Epoch: 54, Loss: 0.75, Accuracy: 0.79, Time: 1077.20 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[858 155  92]\n",
      " [140 884  99]\n",
      " [ 86 103 816]]\n",
      "Break Count: 7\n",
      "Epoch: 55, Loss: 0.75, Accuracy: 0.80, Time: 11140.66 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[912  98  95]\n",
      " [178 871  74]\n",
      " [ 91 129 785]]\n",
      "Break Count: 8\n",
      "Epoch: 56, Loss: 0.75, Accuracy: 0.80, Time: 10406.98 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[936  94  75]\n",
      " [187 860  76]\n",
      " [119 115 771]]\n",
      "Break Count: 9\n",
      "Epoch: 57, Loss: 0.75, Accuracy: 0.80, Time: 9322.14 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[928 104  73]\n",
      " [194 839  90]\n",
      " [111 104 790]]\n",
      "Break Count: 10\n",
      "Epoch: 58, Loss: 0.74, Accuracy: 0.80, Time: 941.39 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[820 216  69]\n",
      " [ 99 985  39]\n",
      " [ 79 149 777]]\n",
      "Break Count: 11\n",
      "Epoch: 59, Loss: 0.74, Accuracy: 0.80, Time: 903.69 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[856 164  85]\n",
      " [137 899  87]\n",
      " [ 84  97 824]]\n",
      "Break Count: 12\n",
      "Epoch: 60, Loss: 0.75, Accuracy: 0.80, Time: 907.09 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.78, Total items: 3233\n",
      "Weighted F1 Score: 0.78\n",
      "Confusion Matrix:\n",
      "[[860 114 131]\n",
      " [163 841 119]\n",
      " [ 96  73 836]]\n",
      "Break Count: 13\n",
      "Epoch: 61, Loss: 0.74, Accuracy: 0.80, Time: 905.45 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[884 156  65]\n",
      " [159 880  84]\n",
      " [ 93 118 794]]\n",
      "Break Count: 14\n",
      "Epoch: 62, Loss: 0.75, Accuracy: 0.80, Time: 1633.55 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[865 118 122]\n",
      " [140 895  88]\n",
      " [ 60 119 826]]\n",
      "Saving model\n",
      "Saving Logs at /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/seed_3class_transition/log/\n",
      "Epoch: 63, Loss: 0.74, Accuracy: 0.80, Time: 913.23 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[884 140  81]\n",
      " [167 897  59]\n",
      " [ 78 116 811]]\n",
      "Break Count: 1\n",
      "Epoch: 64, Loss: 0.74, Accuracy: 0.80, Time: 907.90 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[895 128  82]\n",
      " [191 858  74]\n",
      " [ 95 102 808]]\n",
      "Break Count: 2\n",
      "Epoch: 65, Loss: 0.75, Accuracy: 0.80, Time: 1112.56 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[936 100  69]\n",
      " [203 866  54]\n",
      " [111 129 765]]\n",
      "Break Count: 3\n",
      "Epoch: 66, Loss: 0.74, Accuracy: 0.80, Time: 917.16 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[930 111  64]\n",
      " [184 880  59]\n",
      " [112 122 771]]\n",
      "Break Count: 4\n",
      "Epoch: 67, Loss: 0.74, Accuracy: 0.81, Time: 2055.99 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.80, Total items: 3233\n",
      "Weighted F1 Score: 0.80\n",
      "Confusion Matrix:\n",
      "[[914 112  79]\n",
      " [178 881  64]\n",
      " [ 88 118 799]]\n",
      "Break Count: 5\n",
      "Epoch: 68, Loss: 0.74, Accuracy: 0.81, Time: 689.62 seconds\n",
      "--------------------------------------------------------------------------------\n",
      "Validation\n",
      "Accuracy: 0.79, Total items: 3233\n",
      "Weighted F1 Score: 0.79\n",
      "Confusion Matrix:\n",
      "[[893 133  79]\n",
      " [175 899  49]\n",
      " [107 149 749]]\n",
      "Break Count: 6\n",
      "\u001b[2K\r"
     ]
    }
   ],
   "source": [
    "!python train.py --repo_path /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/ --data_path /Users/davidmcdevitt/Documents/GitHub/speech-pipeline/experiments/seed_3class_transition/data/ --model_name seed_3class_transition --num_epochs 999 --train_split 0.9 --break_count 15 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training --seed_only --transition_class\n"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "lyXvdyRgw5HR"
   },
   "source": [
    "Seed Only 2 Class"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "Y8sQHeb5w5HR"
   },
   "outputs": [],
   "source": [
    "#!python train.py --repo_path /content/drive/MyDrive/Git/speech_pipeline/ --data_path /content/data/ --model_name seed_only_2class --seed_only --num_epochs 999 --train_split 0.9 --break_count 1 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "zu1nTNQksICw"
   },
   "source": [
    "Seed Only + Mixed"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "4sDejW2W92is"
   },
   "outputs": [],
   "source": [
    "#!python train.py --repo_path /content/drive/MyDrive/Git/speech_pipeline/ --data_path /content/data/ --model_name full_2class --num_epochs 999 --train_split 0.9 --break_count 1 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training --overlap_class --seed_only"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "fd6FKCRRsQJg"
   },
   "source": [
    "Full Dataset + Mixed"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "_EPLrmSfsQJh"
   },
   "outputs": [],
   "source": [
    "#!python train.py --repo_path /content/drive/MyDrive/Git/speech_pipeline/ --data_path /content/data/ --model_name full_2class --num_epochs 999 --train_split 0.9 --break_count 1 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training --overlap_class"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "LAumEFiLAnSV"
   },
   "source": [
    "Seed Only + Transitions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "8z2kQM9AAnBu"
   },
   "outputs": [],
   "source": [
    "#!python train.py --repo_path /content/drive/MyDrive/Git/speech_pipeline/ --data_path /content/data/ --model_name full_2class --num_epochs 999 --train_split 0.9 --break_count 1 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training --transition_class --seed_only"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "9IMzunsXNwgd"
   },
   "source": [
    "Full Dataset + Transitions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "rw7HFrn0Nv8R"
   },
   "outputs": [],
   "source": [
    "#!python train.py --repo_path /content/drive/MyDrive/Git/speech_pipeline/ --data_path /content/data/ --model_name full_2class --num_epochs 999 --train_split 0.9 --break_count 1 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training --transition_class"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "KayQfm-i3xNM"
   },
   "source": [
    "Seed Only + Mixed + Transitions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "l8r1KRLnRV2b"
   },
   "outputs": [],
   "source": [
    "#!python train.py --repo_path /content/drive/MyDrive/Git/speech_pipeline/ --data_path /content/data/ --model_name full_2class --num_epochs 999 --train_split 0.9 --break_count 1 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training --transition_class --overlap_class --seed_only"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "id": "IyhDjzgh388C"
   },
   "source": [
    "Full Dataset + Mixed + Transitions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "id": "1c7o-IxD34Yo"
   },
   "outputs": [],
   "source": [
    "#!python train.py --repo_path /content/drive/MyDrive/Git/speech_pipeline/ --data_path /content/data/ --model_name full_2class --num_epochs 999 --train_split 0.9 --break_count 1 --batch_size 16 --clip_size 1 --noise_level 0.2 --class_weights --log_training --transition_class --overlap_class"
   ]
  }
 ],
 "metadata": {
  "colab": {
   "provenance": []
  },
  "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.9.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 1
}