diff --git a/Dockerfile b/Dockerfile index 61f73dc..139b85f 100644 --- a/Dockerfile +++ b/Dockerfile @@ -8,7 +8,7 @@ RUN pip3 install pandas RUN pip3 install kaggle RUN pip3 install tensorflow COPY ./data_train ./ -COPY ./data_test ./ +COPY ./data_dev ./ COPY ./neural_network.sh ./ COPY ./neural_network.py ./ RUN mkdir /.kaggle diff --git a/neural_network.py b/neural_network.py index 5ecd857..495dcca 100644 --- a/neural_network.py +++ b/neural_network.py @@ -37,7 +37,7 @@ model.compile(loss='mean_absolute_error', optimizer="Adam", metrics=['mean_absol model.fit(X, y, epochs=15, validation_split = 0.3) -data = pd.read_csv("data_test", sep=',', error_bad_lines=False).dropna() +data = pd.read_csv("data_dev", sep=',', error_bad_lines=False).dropna() X_test = data.loc[:,data.columns == "2805317"].astype(int) y_test = data.loc[:,data.columns == "198909"].astype(int)