You can train partitioned prediction models from partitioned datasets.
See /machine-learning/partitioned.
