Data visualization project on training data set