nb_iteration in train method