added data_path for train/test; added feature transformation