com.cloudera.oryx.app.schema

Class CategoricalValueEncodings

java.lang.Object

com.cloudera.oryx.app.schema.CategoricalValueEncodings

All Implemented Interfaces:

Serializable

public final class CategoricalValueEncodings
extends Object
implements Serializable

Encapsulates a two-way mapping between the set of values for one categorical feature as strings, and their encodings as integers.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
CategoricalValueEncodings(Map<Integer,? extends Collection<String>> distinctValues)

Method Summary

Modifier and Type	Method and Description
Map<Integer,Integer>	getCategoryCounts()
Map<Integer,String>	getEncodingValueMap(int index)
int	getValueCount(int index)
Map<String,Integer>	getValueEncodingMap(int index)
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

CategoricalValueEncodings

public CategoricalValueEncodings(Map<Integer,? extends Collection<String>> distinctValues)

Parameters:

distinctValues - map of categorical feature indices, to the set of distinct values for that categorical feature. Order matters.

Method Detail

getValueEncodingMap

public Map<String,Integer> getValueEncodingMap(int index)

Parameters:

index - feature index

Returns:

mapping of categorical feature values to numeric encodings for the indicated categorical feature

getEncodingValueMap

public Map<Integer,String> getEncodingValueMap(int index)

Parameters:

index - feature index

Returns:

mapping of numeric encodings to categorical feature values for the indicated categorical feature

getValueCount

public int getValueCount(int index)

Parameters:

index - feature index

Returns:

number of distinct values of the categorical feature

getCategoryCounts

public Map<Integer,Integer> getCategoryCounts()

Returns:

map of categorical feature index to number of distinct values of the feature

toString

public String toString()

Overrides:

toString in class Object