Added error_as_feature to training script