adjust docker file for train and test

This commit is contained in:
filnow 2024-04-28 19:06:29 +02:00
parent ef624e52fd
commit 87a2663e53
3 changed files with 6 additions and 1 deletions

View File

@ -11,5 +11,7 @@ WORKDIR /app
COPY ./download.sh ./ COPY ./download.sh ./
COPY ./calculate.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

View File

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

View File

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