load_data() can now also load predictions in a simpler format.