Class XGBoostClassificationConverter

java.lang.Object

org.tribuo.classification.xgboost.XGBoostClassificationConverter

All Implemented Interfaces:

Serializable, XGBoostOutputConverter<Label>, org.tribuo.protos.ProtoSerializable<org.tribuo.common.xgboost.protos.XGBoostOutputConverterProto>

public final class XGBoostClassificationConverter
extends Object
implements XGBoostOutputConverter<Label>

Converts XGBoost outputs into Label Predictions.

See Also:

• Serialized Form

Field Summary

Fields

Modifier and Type	Field	Description
static final int	CURRENT_VERSION	Protobuf serialization version.

Fields inherited from interface org.tribuo.protos.ProtoSerializable

DESERIALIZATION_METHOD_NAME, PROVENANCE_SERIALIZER

Constructor Summary

Constructors

Constructor	Description
XGBoostClassificationConverter()	Constructs an XGBoostClassificationConverter.

Method Summary

Modifier and Type	Method	Description
List<org.tribuo.Prediction<Label>>	convertBatchOutput(org.tribuo.ImmutableOutputInfo<Label> info, List<float[][]> probabilitiesList, int[] numValidFeatures, org.tribuo.Example<Label>[] examples)
org.tribuo.Prediction<Label>	convertOutput(org.tribuo.ImmutableOutputInfo<Label> info, List<float[]> probabilitiesList, int numValidFeatures, org.tribuo.Example<Label> example)
static XGBoostClassificationConverter	deserializeFromProto(int version, String className, com.google.protobuf.Any message)	Deserialization factory.
boolean	generatesProbabilities()
Class<Label>	getTypeWitness()
org.tribuo.common.xgboost.protos.XGBoostOutputConverterProto	serialize()

Methods inherited from class java.lang.Object

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

Field Details

CURRENT_VERSION

public static final int CURRENT_VERSION

Protobuf serialization version.

See Also:

• Constant Field Values

Constructor Details

XGBoostClassificationConverter

public XGBoostClassificationConverter()

Constructs an XGBoostClassificationConverter.

Method Details

deserializeFromProto

public static XGBoostClassificationConverter deserializeFromProto(int version,
 String className,
 com.google.protobuf.Any message)

Deserialization factory.

Parameters:

version - The serialized object version.

className - The class name.

message - The serialized data.

Returns:

The deserialized object.

generatesProbabilities

public boolean generatesProbabilities()

Specified by:

generatesProbabilities in interface XGBoostOutputConverter<Label>

convertOutput

public org.tribuo.Prediction<Label> convertOutput(org.tribuo.ImmutableOutputInfo<Label> info,
 List<float[]> probabilitiesList,
 int numValidFeatures,
 org.tribuo.Example<Label> example)

Specified by:

convertOutput in interface XGBoostOutputConverter<Label>

convertBatchOutput

public List<org.tribuo.Prediction<Label>> convertBatchOutput(org.tribuo.ImmutableOutputInfo<Label> info,
 List<float[][]> probabilitiesList,
 int[] numValidFeatures,
 org.tribuo.Example<Label>[] examples)

Specified by:

convertBatchOutput in interface XGBoostOutputConverter<Label>

serialize

public org.tribuo.common.xgboost.protos.XGBoostOutputConverterProto serialize()

Specified by:

serialize in interface org.tribuo.protos.ProtoSerializable<org.tribuo.common.xgboost.protos.XGBoostOutputConverterProto>

getTypeWitness

public Class<Label> getTypeWitness()

Specified by:

getTypeWitness in interface XGBoostOutputConverter<Label>