get_data function

A datasource (DS) method to generate training and test sets