CODE EXAMPLE FOR PYTHON

first remove nans then split into train and validation

You should split before pre-processing or imputing.