diff --git a/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/ObjectiveTrait.scala b/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/ObjectiveTrait.scala index f6d7fd496..24e603762 100644 --- a/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/ObjectiveTrait.scala +++ b/jvm-packages/xgboost4j/src/main/scala/ml/dmlc/xgboost4j/scala/ObjectiveTrait.scala @@ -27,7 +27,7 @@ trait ObjectiveTrait extends IObjective { * * @param predicts untransformed margin predicts * @param dtrain training data - * @return List with two float array, correspond to first order grad and second order grad + * @return List with two float array, correspond to grad and hess */ def getGradient(predicts: Array[Array[Float]], dtrain: DMatrix): List[Array[Float]]