Machine learning-assisted railway simulation modelling