{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"provenance":[],"toc_visible":true},"kernelspec":{"name":"python3","display_name":"Python 3"},"language_info":{"name":"python"}},"cells":[{"cell_type":"code","execution_count":null,"metadata":{"id":"7zKdtJlIFhOn"},"outputs":[],"source":["import matplotlib.pyplot as plt\n","import pandas as pd\n","import torch\n","import torch.nn as nn\n","\n","import numpy as np\n","import torch.optim as optim\n","import torch.utils.data as data\n","from sklearn.model_selection import train_test_split\n","from sklearn.preprocessing import MinMaxScaler, LabelEncoder\n","import ast\n","from torch.utils.data import DataLoader, TensorDataset\n","from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score\n","import matplotlib.pyplot as plt"]},{"cell_type":"code","source":["dataset = pd.read_csv('labeled_dataset.csv')"],"metadata":{"id":"MeNmCEtPFzFo"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["dataset.head(15)"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":519},"id":"P9w7w8WWF0Nd","executionInfo":{"status":"ok","timestamp":1692311813192,"user_tz":300,"elapsed":388,"user":{"displayName":"César Mosqueira","userId":"11705195256143475621"}},"outputId":"fde1231c-6d0f-44f8-b547-99975dec11e2"},"execution_count":null,"outputs":[{"output_type":"execute_result","data":{"text/plain":[" video group frame \\\n","0 video_15.mp4 1 5 \n","1 video_15.mp4 1 11 \n","2 video_15.mp4 1 17 \n","3 video_15.mp4 1 23 \n","4 video_15.mp4 1 29 \n","5 video_15.mp4 1 35 \n","6 video_15.mp4 1 41 \n","7 video_15.mp4 1 47 \n","8 video_15.mp4 1 53 \n","9 video_15.mp4 1 59 \n","10 video_15.mp4 2 71 \n","11 video_15.mp4 2 77 \n","12 video_15.mp4 2 83 \n","13 video_15.mp4 2 89 \n","14 video_15.mp4 2 95 \n","\n"," landmarks Label \n","0 [ 334.75 178.55 0.98386 339.... bad \n","1 [ 329.95 181.47 0.99063 334.... bad \n","2 [ 329.7 182.92 0.99079 334... bad \n","3 [ 329.32 187.55 0.98055 334.... bad \n","4 [ 331.31 194.96 0.985 335.... bad \n","5 [ 326.15 199.31 0.98249 330.... bad \n","6 [ 317.77 207.25 0.97823 321... bad \n","7 [ 311.16 194.05 0.97995 315.... bad \n","8 [ 308.41 194.28 0.98219 312.... bad \n","9 [ 318.15 184.29 0.98273 323... bad \n","10 [ 321.08 126 0.99838 332.... bad \n","11 [ 311.02 128.68 0.99853 321.... bad \n","12 [ 301.6 139.2 0.99845 311.... bad \n","13 [ 294.57 158.13 0.9981 303.... bad \n","14 [ 294.98 166.88 0.99792 303.... bad "],"text/html":["\n","
\n"," | video | \n","group | \n","frame | \n","landmarks | \n","Label | \n","
---|---|---|---|---|---|
0 | \n","video_15.mp4 | \n","1 | \n","5 | \n","[ 334.75 178.55 0.98386 339.... | \n","bad | \n","
1 | \n","video_15.mp4 | \n","1 | \n","11 | \n","[ 329.95 181.47 0.99063 334.... | \n","bad | \n","
2 | \n","video_15.mp4 | \n","1 | \n","17 | \n","[ 329.7 182.92 0.99079 334... | \n","bad | \n","
3 | \n","video_15.mp4 | \n","1 | \n","23 | \n","[ 329.32 187.55 0.98055 334.... | \n","bad | \n","
4 | \n","video_15.mp4 | \n","1 | \n","29 | \n","[ 331.31 194.96 0.985 335.... | \n","bad | \n","
5 | \n","video_15.mp4 | \n","1 | \n","35 | \n","[ 326.15 199.31 0.98249 330.... | \n","bad | \n","
6 | \n","video_15.mp4 | \n","1 | \n","41 | \n","[ 317.77 207.25 0.97823 321... | \n","bad | \n","
7 | \n","video_15.mp4 | \n","1 | \n","47 | \n","[ 311.16 194.05 0.97995 315.... | \n","bad | \n","
8 | \n","video_15.mp4 | \n","1 | \n","53 | \n","[ 308.41 194.28 0.98219 312.... | \n","bad | \n","
9 | \n","video_15.mp4 | \n","1 | \n","59 | \n","[ 318.15 184.29 0.98273 323... | \n","bad | \n","
10 | \n","video_15.mp4 | \n","2 | \n","71 | \n","[ 321.08 126 0.99838 332.... | \n","bad | \n","
11 | \n","video_15.mp4 | \n","2 | \n","77 | \n","[ 311.02 128.68 0.99853 321.... | \n","bad | \n","
12 | \n","video_15.mp4 | \n","2 | \n","83 | \n","[ 301.6 139.2 0.99845 311.... | \n","bad | \n","
13 | \n","video_15.mp4 | \n","2 | \n","89 | \n","[ 294.57 158.13 0.9981 303.... | \n","bad | \n","
14 | \n","video_15.mp4 | \n","2 | \n","95 | \n","[ 294.98 166.88 0.99792 303.... | \n","bad | \n","