Merge remote-tracking branch 'origin/master'

This commit is contained in:
Karolina Oparczyk 2021-04-14 20:40:17 +02:00
commit 8e6ce98c78
2 changed files with 13 additions and 1 deletions

12
Dockerfile Normal file
View File

@ -0,0 +1,12 @@
FROM ubuntu:latest
RUN apt update
RUN apt install -y python3
RUN apt install -y python3-pip
RUN apt install -y unzip
RUN pip3 install pandas
RUN pip3 install kaggle
COPY ./get_data.sh ./
COPY ./main.py ./
ENV KAGGLE_USERNAME karopa
ARG key
ENV KAGGLE_KEY $key

View File

@ -16,5 +16,5 @@ if kaggle datasets download -d sgonkaggle/youtube-trend-with-subscriber && unzip
wc -l "data_dev"
echo "Train dataset"
wc -l "data_train"
python main.py USvideos_modified.csv
python3 main.py USvideos_modified.csv
fi