added functionality to train and predict for all Y values - A to G and 
to save the result in csv file