Improve the machine learning components

[janfreyberg]

For the trial-based prediction method, improve what's returned to the user.

For the timepoint-based prediction, update the docs and improve how too much data is handled