org.datavec.api.transform.transform.integer

Class IntegerToOneHotTransform

java.lang.Object

org.datavec.api.transform.transform.BaseTransform

org.datavec.api.transform.transform.integer.IntegerToOneHotTransform

All Implemented Interfaces:

Serializable, ColumnOp, Transform

public class IntegerToOneHotTransform
extends BaseTransform

Convert an integer column to a set of one-hot columns.

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
IntegerToOneHotTransform(String columnName, int minValue, int maxValue)

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 schema) Get the output schema for this transformation, given an input schema

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

IntegerToOneHotTransform

public IntegerToOneHotTransform(String columnName,
                                int minValue,
                                int maxValue)

Method Detail

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

transform

public Schema transform(Schema schema)

Description copied from interface: ColumnOp

Get the output schema for this transformation, given an input schema

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: