Java example source code file (CustomStemmingPreprocessor.java)

This example Java source code file (CustomStemmingPreprocessor.java) is included in the alvinalexander.com "Java Source Code Warehouse" project. The intent of this project is to help you "Learn Java by Example" ^TM.

Learn more about this Java project at its project page.

Java - Java tags/keywords

commonpreprocessor, customstemmingpreprocessor, override, snowballprogram, string

The CustomStemmingPreprocessor.java Java example source code

package org.deeplearning4j.text.tokenization.tokenizer.preprocessor;

import lombok.NonNull;
import org.tartarus.snowball.SnowballProgram;

/**
 * This is StemmingPreprocessor compatible with different StemmingProcessors defined as lucene/tartarus SnowballProgram
 * Like, but not limited to: RussianStemmer, DutchStemmer, FrenchStemmer etc
 *
 * PLEASE NOTE: This preprocessor is NOT thread-safe.
 *
 * @author raver119@gmail.com
 */
public class CustomStemmingPreprocessor extends CommonPreprocessor {
    private SnowballProgram stemmer;
    public CustomStemmingPreprocessor(@NonNull SnowballProgram stemmer) {
        this.stemmer = stemmer;
    }

    @Override
    public String preProcess(String token) {
        String prep = super.preProcess(token);
        stemmer.setCurrent(prep);
        stemmer.stem();
        return stemmer.getCurrent();
    }
}