Loaded train and valid test sets to build sklearn regression model