fixed the wrong cost name bug in model training