Update Dockerfile

Dockerfile

@@ -3,10 +3,8 @@ FROM ubuntu:latest
 RUN apt update && apt install -y python3-pip
 RUN apt install unzip
 RUN apt install bc
-RUN pip install kaggle 
-RUN pip install --user pandas
-RUN pip install --user scikit-learn
-RUN pip install --user matplotlib
+
+RUN pip3 install kaggle pandas scikit-learn
 
 WORKDIR /app
 

IUM_2.py

@@ -1,4 +1,3 @@
-import matplotlib.pyplot as plt 
 import pandas as pd
 import kaggle
 from sklearn.model_selection import train_test_split