Package org.tribuo.classification

Class LabelInfo

java.lang.Object

org.tribuo.classification.LabelInfo

All Implemented Interfaces:

Serializable, OutputInfo<Label>

Direct Known Subclasses:

ImmutableLabelInfo, MutableLabelInfo

public abstract class LabelInfo
extends Object
implements OutputInfo<Label>

The base class for information about multi-class classification Labels.

See Also:

• Serialized Form

Field Summary

Fields

Modifier and Type	Field	Description
protected final Map<String,com.oracle.labs.mlrg.olcut.util.MutableLong>	labelCounts	The occurrence counts of each label.
protected Map<String,Label>	labels	The label domain.
protected int	unknownCount	The number of unknown labels this LabelInfo has seen.

Method Summary

Modifier and Type	Method	Description
abstract LabelInfo	copy()
ImmutableOutputInfo<Label>	generateImmutableOutputInfo()
MutableOutputInfo<Label>	generateMutableOutputInfo()
Set<Label>	getDomain()	Returns the set of possible Labels that this LabelInfo has seen.
long	getLabelCount(String label)	Gets the count of the supplied label, or 0 if the label is unknown.
long	getLabelCount(Label label)	Gets the count of the supplied label, or 0 if the label is unknown.
int	getUnknownCount()
Iterable<com.oracle.labs.mlrg.olcut.util.Pair<String,Long>>	outputCountsIterable()
int	size()	The number of unique Labels this LabelInfo has seen.

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.tribuo.OutputInfo

toReadableString

Field Details

labelCounts

protected final Map<String,com.oracle.labs.mlrg.olcut.util.MutableLong> labelCounts

The occurrence counts of each label.

unknownCount

protected int unknownCount

The number of unknown labels this LabelInfo has seen.

labels

protected transient Map<String,Label> labels

The label domain.

Method Details

getUnknownCount

public int getUnknownCount()

Specified by:

getUnknownCount in interface OutputInfo<Label>

getDomain

public Set<Label> getDomain()

Returns the set of possible Labels that this LabelInfo has seen.

Each label has the default score of Double.NaN.

Specified by:

getDomain in interface OutputInfo<Label>

Returns:

The set of possible labels.

getLabelCount

public long getLabelCount(Label label)

Gets the count of the supplied label, or 0 if the label is unknown.

Parameters:

label - A Label.

Returns:

A non-negative long.

getLabelCount

public long getLabelCount(String label)

Gets the count of the supplied label, or 0 if the label is unknown.

Parameters:

label - A String representing a Label.

Returns:

A non-negative long.

outputCountsIterable

public Iterable<com.oracle.labs.mlrg.olcut.util.Pair<String,Long>> outputCountsIterable()

Specified by:

outputCountsIterable in interface OutputInfo<Label>

size

public int size()

The number of unique Labels this LabelInfo has seen.

Specified by:

size in interface OutputInfo<Label>

Returns:

The number of unique labels.

generateImmutableOutputInfo

public ImmutableOutputInfo<Label> generateImmutableOutputInfo()

Specified by:

generateImmutableOutputInfo in interface OutputInfo<Label>

generateMutableOutputInfo

public MutableOutputInfo<Label> generateMutableOutputInfo()

Specified by:

generateMutableOutputInfo in interface OutputInfo<Label>

copy

public abstract LabelInfo copy()

Specified by:

copy in interface OutputInfo<Label>