org.datavec.api.transform.ndarray

Class NDArrayDistanceTransform

java.lang.Object

org.datavec.api.transform.ndarray.NDArrayDistanceTransform

All Implemented Interfaces:

Serializable, ColumnOp, Transform

public class NDArrayDistanceTransform
extends Object
implements Transform

Calculate the distance (cosine similarity, EUCLIDEAN, MANHATTAN) between two INDArrays

Author:

Alex Black

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
NDArrayDistanceTransform(String newColumnName, Distance distance, String firstCol, String secondCol)

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
List<List<Writable>>	mapSequence(List<List<Writable>> sequence) Transform a sequence
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) Get the output schema for this transformation, given an input schema

Methods inherited from class java.lang.Object

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

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

getInputSchema

Constructor Detail

NDArrayDistanceTransform

public NDArrayDistanceTransform(@NonNull
                                String newColumnName,
                                @NonNull
                                Distance distance,
                                @NonNull
                                String firstCol,
                                @NonNull
                                String secondCol)

Method Detail

toString

public String toString()

Overrides:

toString in class Object

setInputSchema

public void setInputSchema(Schema inputSchema)

Description copied from interface: ColumnOp

Set the input schema.

Specified by:

setInputSchema in interface ColumnOp

map

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

Description copied from interface: Transform

Transform a writable in to another writable

Specified by:

map in interface Transform

Parameters:

writables - the record to transform

Returns:

the transformed writable

mapSequence

public List<List<Writable>> mapSequence(List<List<Writable>> sequence)

Description copied from interface: Transform

Transform a sequence

Specified by:

mapSequence in interface Transform

map

public Object map(Object input)

Description copied from interface: Transform

Transform an object in to another object

Specified by:

map in interface Transform

Parameters:

input - the record to transform

Returns:

the transformed writable

mapSequence

public Object mapSequence(Object sequence)

Description copied from interface: Transform

Transform a sequence

Specified by:

mapSequence in interface Transform

transform

public Schema transform(Schema inputSchema)

Description copied from interface: ColumnOp

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

Specified by:

transform in interface ColumnOp

outputColumnName

public String outputColumnName()

Description copied from interface: ColumnOp

The output column name after the operation has been applied

Specified by:

outputColumnName in interface ColumnOp

Returns:

the output column name

outputColumnNames

public String[] outputColumnNames()

Description copied from interface: ColumnOp

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

Specified by:

outputColumnNames in interface ColumnOp

Returns:

the output column names

columnNames

public String[] columnNames()

Description copied from interface: ColumnOp

Returns column names this op is meant to run on

Specified by:

columnNames in interface ColumnOp

Returns:

columnName

public String columnName()

Description copied from interface: ColumnOp

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

Specified by:

columnName in interface ColumnOp

Returns: