org.datavec.api.transform.transform.categorical

Class CategoricalToOneHotTransform

java.lang.Object

org.datavec.api.transform.transform.BaseTransform

org.datavec.api.transform.transform.categorical.CategoricalToOneHotTransform

All Implemented Interfaces:

Serializable, ColumnOp, Transform

public class CategoricalToOneHotTransform
extends BaseTransform

Created by Alex on 4/03/2016.

See Also:

Serialized Form

Field Summary

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

inputSchema

Constructor Summary

Constructors

Constructor and Description
CategoricalToOneHotTransform(String columnName)

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
boolean	equals(Object o)
int	hashCode()
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, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

CategoricalToOneHotTransform

public CategoricalToOneHotTransform(String columnName)

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

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object

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: