adjust docker file for train and test

Dockerfile

@@ -11,5 +11,7 @@ WORKDIR /app
 
 COPY ./download.sh ./
 COPY ./calculate.sh ./
+COPY ./train.py ./
+COPY ./test.py ./
 
-CMD bash
+CMD bash download.sh && bash calculate.sh && python train.py && python test.py

download.sh

@@ -2,4 +2,5 @@ pip install kaggle
 kaggle datasets download -d fanconic/skin-cancer-malignant-vs-benign
 unzip skin-cancer-malignant-vs-benign.zip 
 rm skin-cancer-malignant-vs-benign.zip
+rm -r data
 

requirements.txt

@@ -1,3 +1,5 @@
 matplotlib==3.8.3
 numpy==1.26.4
+torchvision==0.17.1
+torch==2.2.1