When I saw the NaNs, my first thought was the example was going to be predict the NaN values, not drop them.