Сначала удалите NAN, а затем разделитесь на поезд и проверку

You should split before pre-processing or imputing.
Real Raccoon