Clean up documentation of Net.
Move code to set training and validation set to learner.