nan error fix v3

This commit is contained in:
Karolina Oparczyk 2021-05-17 19:27:39 +02:00
parent 83361bdf43
commit a41ab2fae1
2 changed files with 2 additions and 2 deletions

View File

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

View File

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