Adding the logic to make models and batch predictions form split datasets