Class FirstDigitTransform
- java.lang.Object
-
- org.datavec.api.transform.transform.BaseTransform
-
- org.datavec.api.transform.transform.categorical.FirstDigitTransform
-
public class FirstDigitTransform extends BaseTransform
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
FirstDigitTransform.Mode
Mode determines how non-numerical entries should be handled:
EXCEPTION_ON_INVALID: output has 10 category values ("0", ..., "9"), and any non-numerical values result in an exception
INCLUDE_OTHER_CATEGORY: output has 11 category values ("0", ..., "9", "Other"), all non-numerical values are mapped to "Other"
-
Field Summary
Fields Modifier and Type Field Description protected String
inputColumn
protected FirstDigitTransform.Mode
mode
static String
OTHER_CATEGORY
protected String
outputColumn
-
Fields inherited from class org.datavec.api.transform.transform.BaseTransform
inputSchema
-
-
Constructor Summary
Constructors Constructor Description FirstDigitTransform(String inputColumn, String outputColumn, FirstDigitTransform.Mode mode)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description String
columnName()
Returns a singular column name this op is meant to run onString[]
columnNames()
Returns column names this op is meant to run onObject
map(Object input)
Transform an object in to another objectList<Writable>
map(List<Writable> writables)
Transform a writable in to another writableObject
mapSequence(Object sequence)
Transform a sequenceString
outputColumnName()
The output column name after the operation has been appliedString[]
outputColumnNames()
The output column names This will often be the same as the inputvoid
setInputSchema(Schema schema)
Set the input schema.String
toString()
Schema
transform(Schema inputSchema)
-
Methods inherited from class org.datavec.api.transform.transform.BaseTransform
getInputSchema, mapSequence
-
-
-
-
Field Detail
-
OTHER_CATEGORY
public static final String OTHER_CATEGORY
- See Also:
- Constant Field Values
-
inputColumn
protected String inputColumn
-
outputColumn
protected String outputColumn
-
mode
protected FirstDigitTransform.Mode mode
-
-
Constructor Detail
-
FirstDigitTransform
public FirstDigitTransform(String inputColumn, String outputColumn, FirstDigitTransform.Mode mode)
- Parameters:
inputColumn
- Input column nameoutputColumn
- Output column name. If same as input, input column is replacedmode
- SeeFirstDigitTransform.Mode
-
-
Method Detail
-
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)
Description copied from interface:Transform
Transform an object in to another object- Parameters:
input
- the record to transform- Returns:
- the transformed writable
-
mapSequence
public Object mapSequence(Object sequence)
Description copied from interface:Transform
Transform a sequence
-
toString
public String toString()
- Specified by:
toString
in classBaseTransform
-
outputColumnName
public String outputColumnName()
Description copied from interface:ColumnOp
The output column name after the operation has been applied- 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- Returns:
- the output column names
-
columnNames
public String[] columnNames()
Description copied from interface:ColumnOp
Returns column names this op is meant to run on- Returns:
-
columnName
public String columnName()
Description copied from interface:ColumnOp
Returns a singular column name this op is meant to run on- Returns:
-
setInputSchema
public void setInputSchema(Schema schema)
Description copied from interface:ColumnOp
Set the input schema.- Specified by:
setInputSchema
in interfaceColumnOp
- Overrides:
setInputSchema
in classBaseTransform
-
-