org.tribuo.common.xgboost

Enum XGBoostTrainer.BoosterType

java.lang.Object

java.lang.Enum<XGBoostTrainer.BoosterType>

org.tribuo.common.xgboost.XGBoostTrainer.BoosterType

All Implemented Interfaces:

Serializable, Comparable<XGBoostTrainer.BoosterType>

Enclosing class:

XGBoostTrainer<T extends Output<T>>

public static enum XGBoostTrainer.BoosterType
extends Enum<XGBoostTrainer.BoosterType>

The type of XGBoost model.

Enum Constant Summary

Enum Constants

Enum Constant and Description
DART A gradient boosted decision tree using dropout.
GBTREE A gradient boosted decision tree.
LINEAR A boosted linear model.

Field Summary

Fields

Modifier and Type	Field and Description
String	paramName

Method Summary

Modifier and Type	Method and Description
static XGBoostTrainer.BoosterType	valueOf(String name) Returns the enum constant of this type with the specified name.
static XGBoostTrainer.BoosterType[]	values() Returns an array containing the constants of this enum type, in the order they are declared.

Methods inherited from class java.lang.Enum

clone, compareTo, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

Enum Constant Detail

LINEAR

public static final XGBoostTrainer.BoosterType LINEAR

A boosted linear model.

GBTREE

public static final XGBoostTrainer.BoosterType GBTREE

A gradient boosted decision tree.

DART

public static final XGBoostTrainer.BoosterType DART

A gradient boosted decision tree using dropout.

Field Detail

paramName

public final String paramName

Method Detail

values

public static XGBoostTrainer.BoosterType[] values()

Returns an array containing the constants of this enum type, in the order they are declared. This method may be used to iterate over the constants as follows:

for (XGBoostTrainer.BoosterType c : XGBoostTrainer.BoosterType.values())
    System.out.println(c);

Returns:

an array containing the constants of this enum type, in the order they are declared

valueOf

public static XGBoostTrainer.BoosterType valueOf(String name)

Returns the enum constant of this type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (Extraneous whitespace characters are not permitted.)

Parameters:

name - the name of the enum constant to be returned.

Returns:

the enum constant with the specified name

Throws:

IllegalArgumentException - if this enum type has no constant with the specified name

NullPointerException - if the argument is null