Ruifeng Zheng created SPARK-59038:
-------------------------------------
Summary: Avoid broadcasting regression tree models for no-op
transforms
Key: SPARK-59038
URL: https://issues.apache.org/jira/browse/SPARK-59038
Project: Spark
Issue Type: Sub-task
Components: ML
Affects Versions: 5.0.0
Reporter: Ruifeng Zheng
RandomForestRegressionModel and GBTRegressionModel eagerly broadcast the model
before determining that both predictionCol and leafCol are empty. Avoid
constructing the broadcast when transform has no output columns.
--
This message was sent by Atlassian Jira
(v8.20.10#820010)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
