Update fertility-prediction.Rmd

esciencecenter-digital-skills · Jun 19, 2024 · c04dabb · c04dabb
1 parent af4155c
commit c04dabb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/episodes/fertility-prediction.Rmd b/episodes/fertility-prediction.Rmd
@@ -295,8 +295,8 @@ How many samples do we have left?
 # The rows where any of the features is NaN
 X_isna = features.isna().any(axis=1)
 
-features = features.drop(features[y_isna].index)
-outcome = outcome.drop(outcome[y_isna].index)
+features = features.drop(features[X_isna].index)
+outcome = outcome.drop(outcome[X_isna].index)
 ```
 
 ```python