ws.palladian.classification.xgboost

Class XGBoostLearner

java.lang.Object

ws.palladian.core.AbstractLearner<XGBoostModel>

ws.palladian.classification.xgboost.XGBoostLearner

All Implemented Interfaces:

Learner<XGBoostModel>

public class XGBoostLearner
extends AbstractLearner<XGBoostModel>

Learner for XGBoost: "Scalable and Flexible Gradient Boosting".

Author:

pk

Constructor Summary

Constructors

Constructor and Description
XGBoostLearner() Create a new XGBoostModel with settings which I took from some Kaggle posts.
XGBoostLearner(Map<String,Object> params, int rounds) Create a new XGBoostLearner instance with the supplied parameters.
XGBoostLearner(Map<String,Object> params, int rounds, ml.dmlc.xgboost4j.java.IEvaluation evaluation) Create a new XGBoostLearner instance with the supplied parameters and a custom evaluation function.

Method Summary

Modifier and Type	Method and Description
XGBoostModel	train(Dataset dataset)
XGBoostModel	train(Dataset training, Dataset validation)

Methods inherited from class ws.palladian.core.AbstractLearner

toString, train

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

XGBoostLearner

public XGBoostLearner(Map<String,Object> params,
                      int rounds,
                      ml.dmlc.xgboost4j.java.IEvaluation evaluation)

Create a new XGBoostLearner instance with the supplied parameters and a custom evaluation function.

Parameters:

params - The parameters, see XGBoost Parameters for a description of all available parameters.

rounds - The number of rounds for boosting.

evaluation - The evaluation function.

XGBoostLearner

public XGBoostLearner(Map<String,Object> params,
                      int rounds)

Create a new XGBoostLearner instance with the supplied parameters.

Parameters:

params - The parameters, see XGBoost Parameters for a description of all available parameters.

rounds - The number of rounds for boosting.

XGBoostLearner

public XGBoostLearner()

Create a new XGBoostModel with settings which I took from some Kaggle posts.

Method Detail

train

public XGBoostModel train(Dataset training,
                          Dataset validation)

Specified by:

train in interface Learner<XGBoostModel>

Overrides:

train in class AbstractLearner<XGBoostModel>

train

public XGBoostModel train(Dataset dataset)