Spark ML – Pipeline stage extraction
R/ml_pipeline_utils.R
ml_stage
Description
Extraction of stages from a Pipeline or PipelineModel object.
Usage
ml_stage(x, stage)
ml_stages(x, stages = NULL) Arguments
| Arguments | Description |
|---|---|
| x | A ml_pipeline or a ml_pipeline_model object |
| stage | The UID of a stage in the pipeline. |
| stages | The UIDs of stages in the pipeline as a character vector. |
Value
For ml_stage(): The stage specified.
For ml_stages(): A list of stages. If stages is not set, the function returns all stages of the pipeline in a list.