dataiku
dataikuapi
You can train partitioned prediction models from partitioned datasets.
See Partitioned Models.