Abstract away data partitioning
Move training out of #run and into a method
Move calling the classifying lambda out of #run