Train model cleaning