The analysis stuff features some task specific steps (eg reconstruction using the classification image datasets)
For reusability across tasks (in the rl setting), it would be nice if those were optional.