Sample script for performing model training and testing