s464915/ium_464915
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

adjust docker file for train and test

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Dockerfile
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

1
download.sh
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

2
requirements.txt
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