Package ai.djl.modality.nlp.preprocess
Class TextTruncator
- java.lang.Object
-
- ai.djl.modality.nlp.preprocess.TextTruncator
-
- All Implemented Interfaces:
TextProcessor
public class TextTruncator extends java.lang.Object implements TextProcessor
TextProcessor
that truncates text to a maximum size.
-
-
Constructor Summary
Constructors Constructor Description TextTruncator(int maxSize)
Constructs aTextTruncator
.
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description java.util.List<java.lang.String>
preprocess(java.util.List<java.lang.String> tokens)
Applies the preprocessing defined to the given input tokens.
-
-
-
Constructor Detail
-
TextTruncator
public TextTruncator(int maxSize)
Constructs aTextTruncator
.- Parameters:
maxSize
- the size to limit the text to
-
-
Method Detail
-
preprocess
public java.util.List<java.lang.String> preprocess(java.util.List<java.lang.String> tokens)
Applies the preprocessing defined to the given input tokens.- Specified by:
preprocess
in interfaceTextProcessor
- Parameters:
tokens
- the tokens created after the input text is tokenized- Returns:
- the preprocessed tokens
-
-