org.datavec.api.transform.sequence.split

Class SplitMaxLengthSequence

java.lang.Object

org.datavec.api.transform.sequence.split.SplitMaxLengthSequence

All Implemented Interfaces:

Serializable, SequenceSplit

public class SplitMaxLengthSequence
extends Object
implements SequenceSplit

Split a sequence into a number of smaller sequences of length 'maxSequenceLength'. If the sequence length is smaller than maxSequenceLength, the sequence is unchanged

Author:

Alex Black

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
SplitMaxLengthSequence(int maxSequenceLength, boolean equalSplits)

Method Summary

Modifier and Type	Method and Description
Schema	getInputSchema() Getter for the input schema
void	setInputSchema(Schema inputSchema) Sets the input schema for this split
List<List<List<Writable>>>	split(List<List<Writable>> sequence) Split a sequence in to multiple time steps
String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

SplitMaxLengthSequence

public SplitMaxLengthSequence(int maxSequenceLength,
                              boolean equalSplits)

Parameters:

maxSequenceLength - max length of sequences

equalSplits - if true: split larger sequences into equal sized subsequences. If false: split into n maxSequenceLength sequences, and (if necessary) 1 with 1 <= length < maxSequenceLength

Method Detail

split

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

Description copied from interface: SequenceSplit

Split a sequence in to multiple time steps

Specified by:

split in interface SequenceSplit

Parameters:

sequence - the sequence to split

Returns:

setInputSchema

public void setInputSchema(Schema inputSchema)

Description copied from interface: SequenceSplit

Sets the input schema for this split

Specified by:

setInputSchema in interface SequenceSplit

Parameters:

inputSchema - the schema to set

getInputSchema

public Schema getInputSchema()

Description copied from interface: SequenceSplit

Getter for the input schema

Specified by:

getInputSchema in interface SequenceSplit

Returns:

toString

public String toString()

Overrides:

toString in class Object