ml.dmlc.xgboost4j.java

Interface IObjective

All Superinterfaces:

Serializable

public interface IObjective
extends Serializable

interface for customize Object function

Author:

hzx

Method Summary

Modifier and Type	Method and Description
List<float[]>	getGradient(float[][] predicts, DMatrix dtrain) user define objective function, return gradient and second order gradient

Method Detail

getGradient

List<float[]> getGradient(float[][] predicts,
                          DMatrix dtrain)

user define objective function, return gradient and second order gradient

Parameters:

predicts - untransformed margin predicts

dtrain - training data

Returns:

List with two float array, correspond to first order grad and second order grad