finished CNN, good accruacy overall but not good for minority and harder classes

.gitignore

@@ -193,3 +193,8 @@ ipython_config.py
 data/**
 !data/**/
 !**/.gitkeep
+
+
+
+models/**
+!**/.gitkeep

requirements.txt

@@ -1,7 +1,9 @@
-appnope==0.1.4
 asttokens==3.0.1
 comm==0.2.3
 contourpy==1.3.3
+cuda-bindings==12.9.4
+cuda-pathfinder==1.2.2
+cuda-toolkit==12.8.1
 cycler==0.12.1
 debugpy==1.8.20
 decorator==5.2.1
@@ -14,19 +16,32 @@ ipython==9.10.0
 ipython_pygments_lexers==1.1.1
 jedi==0.19.2
 Jinja2==3.1.6
-joblib==1.5.3
 jupyter_client==8.8.0
 jupyter_core==5.9.1
 kiwisolver==1.5.0
-MarkupSafe==3.0.3
+MarkupSafe==3.0.2
 matplotlib==3.10.8
 matplotlib-inline==0.2.1
 mpmath==1.3.0
 nest-asyncio==1.6.0
 networkx==3.6.1
 numpy==2.4.3
+nvidia-cublas-cu12==12.8.4.1
+nvidia-cuda-cupti-cu12==12.8.90
+nvidia-cuda-nvrtc-cu12==12.8.93
+nvidia-cuda-runtime-cu12==12.8.90
+nvidia-cudnn-cu12==9.20.0.48
+nvidia-cufft-cu12==11.3.3.83
+nvidia-cufile-cu12==1.13.1.3
+nvidia-curand-cu12==10.3.9.90
+nvidia-cusolver-cu12==11.7.3.90
+nvidia-cusparse-cu12==12.5.8.93
+nvidia-cusparselt-cu12==0.7.1
+nvidia-nccl-cu12==2.29.7
+nvidia-nvjitlink-cu12==12.8.93
+nvidia-nvshmem-cu12==3.4.5
+nvidia-nvtx-cu12==12.8.90
 packaging==26.0
-pandas==3.0.1
 parso==0.8.6
 pexpect==4.9.0
 pillow==12.1.1
@@ -39,16 +54,14 @@ Pygments==2.19.2
 pyparsing==3.3.2
 python-dateutil==2.9.0.post0
 pyzmq==27.1.0
-scikit-learn==1.8.0
-scipy==1.17.1
-seaborn==0.13.2
 six==1.17.0
 stack-data==0.6.3
 sympy==1.14.0
-threadpoolctl==3.6.0
+torch==2.12.0.dev20260318+cu128
-torch==2.10.0
+torchaudio==2.11.0.dev20260318+cu128
-torchvision==0.25.0
+torchvision==0.26.0.dev20260318+cu128
 tornado==6.5.5
 traitlets==5.14.3
+triton==3.6.0+git9844da95
 typing_extensions==4.15.0
 wcwidth==0.6.0