s434766/ium_434766
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix dockerffile

Browse Source
This commit is contained in:
s434766 2021-06-12 11:49:08 +02:00
parent 88f5323fbf
commit 9289cb8a82
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Dockerfile
View File

@ -1,4 +1,4 @@
FROM ubuntu:latest FROM ubuntu:20.04
RUN apt-get update && apt-get install -y python3-pip && pip3 install setuptools && pip3 install numpy && pip3 install pandas && pip3 install wget && pip3 install scikit-learn && pip3 install matplotlib && rm -rf /var/lib/apt/lists/* RUN apt-get update && apt-get install -y python3-pip && pip3 install setuptools && pip3 install numpy && pip3 install pandas && pip3 install wget && pip3 install scikit-learn && pip3 install matplotlib && rm -rf /var/lib/apt/lists/*
RUN pip3 install torch torchvision torchaudio RUN pip3 install torch torchvision torchaudio

3
dvc.yaml
View File

@ -1,9 +1,8 @@
stages: stages:
download_and_split: download_and_split:
cmd: python3 split_10.py cmd: python3 create.py
deps: deps:
- healthcare-dataset-stroke-data.csv - healthcare-dataset-stroke-data.csv
- create.py
outs: outs:
- data_train.csv - data_train.csv
- data_test.csv - data_test.csv