method to build training pipeline for sentiment analysis with Spark NLP

getSentimentTrainingPipeline.java

private Pipeline getSentimentTrainingPipeline() {
    DocumentAssembler document = new DocumentAssembler();
    document.setInputCol("text");
    document.setOutputCol("document");

    String[] tokenizerInputCols = {"document"};
    Tokenizer tokenizer = new Tokenizer();
    tokenizer.setInputCols(tokenizerInputCols);
    tokenizer.setOutputCol("token");

    String[] sentimentInputCols = {"document", "token"};
    ViveknSentimentApproach sentimentApproach = new ViveknSentimentApproach();
    sentimentApproach.setInputCols(sentimentInputCols);
    sentimentApproach.setOutputCol("sentiment");
    sentimentApproach.setSentimentCol("label");
    sentimentApproach.setCorpusPrune(0);

    Pipeline pipeline = new Pipeline();
    pipeline.setStages(new PipelineStage[]{document, tokenizer, sentimentApproach});
    return pipeline;
}