the graphs now show projection for test set instead of train set