Docker image

Dockerfile

@@ -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

get_data.sh

@@ -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