train_MLP_stress.py