Update trainAlgorithm.h