update get_data_sklearn_format, move from training to utils