ml_stage: Spark ML -- Pipeline stage extraction
Description
Extraction of stages from a Pipeline or PipelineModel object.
Usage
ml_stage(x, stage)ml_stages(x, stages = NULL)
Arguments
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.