org.datavec.api.transform.transform.categorical

Class StringToCategoricalTransform

java.lang.Object

org.datavec.api.transform.transform.BaseTransform

org.datavec.api.transform.transform.BaseColumnTransform

org.datavec.api.transform.transform.categorical.StringToCategoricalTransform

All Implemented Interfaces:

Serializable, ColumnOp, Transform

public class StringToCategoricalTransform
extends BaseColumnTransform

Convert a String column to a categorical column

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
StringToCategoricalTransform(String columnName, List<String> stateNames)
StringToCategoricalTransform(String columnName, String... stateNames)

Method Summary

Modifier and Type	Method and Description
ColumnMetaData	getNewColumnMetaData(String newColumnName, ColumnMetaData oldColumnType)
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, equals, hashCode, 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

StringToCategoricalTransform

public StringToCategoricalTransform(String columnName,
                                    List<String> stateNames)

StringToCategoricalTransform

public StringToCategoricalTransform(String columnName,
                                    String... stateNames)

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

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 -