org.datavec.api.transform.transform.categorical

Class IntegerToCategoricalTransform

java.lang.Object

org.datavec.api.transform.transform.BaseTransform

org.datavec.api.transform.transform.BaseColumnTransform

org.datavec.api.transform.transform.categorical.IntegerToCategoricalTransform

All Implemented Interfaces:

Serializable, ColumnOp, Transform

public class IntegerToCategoricalTransform
extends BaseColumnTransform

Convert an integer column to a categorical column, using a provided Map<Integer,String>

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields inherited from class org.datavec.api.transform.transform.BaseColumnTransform

columnName, columnNumber

Fields inherited from class org.datavec.api.transform.transform.BaseTransform

inputSchema

Constructor Summary

Constructors

Constructor and Description
IntegerToCategoricalTransform(String columnName, List<String> list)
IntegerToCategoricalTransform(String columnName, Map<Integer,String> map)

Method Summary

Modifier and Type	Method and Description
boolean	equals(Object o)
ColumnMetaData	getNewColumnMetaData(String newColumnName, ColumnMetaData oldColumnType)
int	hashCode()
Object	map(Object input) Transform an object in to another object
Writable	map(Writable columnWritable)
Object	mapSequence(Object sequence) Transform a sequence
String	toString()

Methods inherited from class org.datavec.api.transform.transform.BaseColumnTransform

columnName, columnNames, map, outputColumnName, outputColumnNames, setInputSchema, transform

Methods inherited from class org.datavec.api.transform.transform.BaseTransform

getInputSchema, mapSequence

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Methods inherited from interface org.datavec.api.transform.ColumnOp

getInputSchema

Constructor Detail

IntegerToCategoricalTransform

public IntegerToCategoricalTransform(String columnName,
                                     Map<Integer,String> map)

IntegerToCategoricalTransform

public IntegerToCategoricalTransform(String columnName,
                                     List<String> list)

Method Detail

getNewColumnMetaData

public ColumnMetaData getNewColumnMetaData(String newColumnName,
                                           ColumnMetaData oldColumnType)

Specified by:

getNewColumnMetaData in class BaseColumnTransform

map

public Writable map(Writable columnWritable)

Specified by:

map in class BaseColumnTransform

toString

public String toString()

Specified by:

toString in class BaseColumnTransform

equals

public boolean equals(Object o)

Overrides:

equals in class BaseColumnTransform

hashCode

public int hashCode()

Overrides:

hashCode in class BaseColumnTransform

map

public Object map(Object input)

Transform an object in to another object

Parameters:

input - the record to transform

Returns:

the transformed writable

mapSequence

public Object mapSequence(Object sequence)

Transform a sequence

Specified by:

mapSequence in interface Transform

Overrides:

mapSequence in class BaseColumnTransform

Parameters:

sequence -