org.datavec.api.transform.transform.string

Class StringListToCategoricalSetTransform

java.lang.Object

org.datavec.api.transform.transform.BaseTransform

org.datavec.api.transform.transform.string.StringListToCategoricalSetTransform

All Implemented Interfaces:

Serializable, ColumnOp, Operation<Schema,Schema>, Transform

public class StringListToCategoricalSetTransform
extends BaseTransform

Convert a delimited String to a list of binary categorical columns. Suppose the possible String values were {"a","b","c","d"} and the String column value to be converted contained the String "a,c", then the 4 output columns would have values ["true","false","true","false"]

Author:

Alex Black

See Also:

Serialized Form

Field Summary

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

inputSchema

Constructor Summary

Constructors

Constructor and Description
StringListToCategoricalSetTransform(String columnName, List<String> newColumnNames, List<String> categoryTokens, String delimiter)

Method Summary

Modifier and Type	Method and Description
String	columnName() Returns a singular column name this op is meant to run on
String[]	columnNames() Returns column names this op is meant to run on
List<Writable>	map(List<Writable> writables) Transform a writable in to another writable
Object	map(Object input) Transform an object in to another object
Object	mapSequence(Object sequence) Transform a sequence
String	outputColumnName() The output column name after the operation has been applied
String[]	outputColumnNames() The output column names This will often be the same as the input
void	setInputSchema(Schema inputSchema) Set the input schema.
String	toString()
Schema	transform(Schema inputSchema)

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

getInputSchema, mapSequence

Methods inherited from class java.lang.Object

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

Constructor Detail

StringListToCategoricalSetTransform

public StringListToCategoricalSetTransform(String columnName,
                                           List<String> newColumnNames,
                                           List<String> categoryTokens,
                                           String delimiter)

Parameters:

columnName - The name of the column to convert

newColumnNames - The names of the new columns to create

categoryTokens - The possible tokens that may be present. Note this list must have the same length and order as the newColumnNames list

delimiter - The delimiter for the Strings to convert

Method Detail

transform

public Schema transform(Schema inputSchema)

setInputSchema

public void setInputSchema(Schema inputSchema)

Description copied from interface: ColumnOp

Set the input schema.

Specified by:

setInputSchema in interface ColumnOp

Overrides:

setInputSchema in class BaseTransform

toString

public String toString()

Specified by:

toString in class BaseTransform

map

public List<Writable> map(List<Writable> writables)

Description copied from interface: Transform

Transform a writable in to another writable

Parameters:

writables - the record to transform

Returns:

the transformed writable

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

Parameters:

sequence -

outputColumnName

public String outputColumnName()

The output column name after the operation has been applied

Returns:

the output column name

outputColumnNames

public String[] outputColumnNames()

The output column names This will often be the same as the input

Returns:

the output column names

columnNames

public String[] columnNames()

Returns column names this op is meant to run on

Returns:

columnName

public String columnName()

Returns a singular column name this op is meant to run on

Returns: