Adding the logic to make models and batch predictions form split datasets
Adding tests and some docs comments for batch predictions with test split