upd51

2023-05-14 22:14:39 +02:00 · 2023-05-14 22:14:39 +02:00 · c3b7f91d57
commit c3b7f91d57
parent 9d0113448c
1 changed files with 2 additions and 1 deletions
--- a/script5_1.py
+++ b/script5_1.py
@ -56,7 +56,8 @@ y_train.fillna(y_train.mean(), inplace=True)
 y_train = y_train.values

 # Filter out non-numeric and NaN values from y_train
-y_train = np.array([x for x in y_train if np.isfinite(x)])
+y_train = y_train.astype(float)
+y_train = y_train[np.isfinite(y_train)]

 # Create the neural network model
 model = Sequential()