clean up after risk model training