bug in IterativeTraining_1x.py line 476, prob is not defined, just in the loop
bug in IterativeTraining_1x.py line 476, prob is not defined, just in the loop