adding train/test split to outlier regression