How to create a data set for machine learning with limited data