nlp_marian_transformer: Spark NLP MarianTransformer

In r-spark/sparknlp: R Interface to John Snow Labs Spark NLP

Spark NLP MarianTransformer

Description

Spark ML transformer that See https://nlp.johnsnowlabs.com/api/#com.johnsnowlabs.nlp.annotators.seq2seq.MarianTransformer

Usage

nlp_marian_transformer(
  x,
  input_cols,
  output_col,
  lang_id = NULL,
  max_input_length = NULL,
  max_output_length = NULL,
  vocabulary = NULL,
  uid = random_string("marian_transformer_")
)

Arguments

x	A spark_connection, ml_pipeline, or a tbl_spark.
input_cols	Input columns. String array.
output_col	Output column. String.
lang_id	A string representing the target language in the form of >>id<< (id = valid target language ID)
max_input_length	Controls the maximum length for encoder inputs (source language texts) Default: 40
max_output_length	Controls the maximum length for decoder outputs (target language texts) Default: 40
vocabulary	Vocabulary used to encode and decode piece tokens generated by SentencePiece This will be set once the model is created and cannot be changed afterwards
uid	A character string used to uniquely identify the ML estimator.

Value

The object returned depends on the class of x.

• spark_connection: When x is a spark_connection, the function returns an instance of a ml_estimator object. The object contains a pointer to a Spark Estimator object and can be used to compose Pipeline objects.

• ml_pipeline: When x is a ml_pipeline, the function returns a ml_pipeline with the NLP estimator appended to the pipeline.

• tbl_spark: When x is a tbl_spark, an estimator is constructed then immediately fit with the input tbl_spark, returning an NLP model.

r-spark/sparknlp documentation built on Oct. 15, 2022, 10:50 a.m.