augment script for run for all 100% training set