make_folds_multi function

Create folds for multi-task and transfer learning