diff --git a/Dockerfile b/Dockerfile index 9b90036..434bc94 100644 --- a/Dockerfile +++ b/Dockerfile @@ -13,7 +13,7 @@ RUN pip3 install pandas RUN pip3 install scikit-learn RUN pip3 install matplotlib RUN mkdir ~/.kaggle/ -#RUN echo '{"username":"riraasaa","key":"1b1376b538ecd7da9e79b94d218ae3ec"}' > ~/.kaggle/kaggle.json + RUN sh chmod a+x ./script.sh # Create app directory in image WORKDIR /app diff --git a/data_processing.py b/data_processing.py index c859a96..c9a131a 100644 --- a/data_processing.py +++ b/data_processing.py @@ -8,7 +8,7 @@ wine = pd.read_csv('winequality-red.csv') X_train,X_rem,y_train,y_rem = train_test_split(wine.iloc[:,:-1],wine.iloc[:,-1], test_size=0.2, random_state=1,stratify=wine["quality"]) X_valid, X_test, y_valid, y_test = train_test_split(X_rem,y_rem, test_size=0.5) -print("Wielkosc danych: train,test,valid:") +print("Wielkosc danych: train, test, valid:") print(X_train.shape) print(X_valid.shape) print(X_test.shape)