Fix bug in the classification, and added learning curve graph.