Split the full training dataset into 2 partitions for 2 diff wombats