[
https://issues.apache.org/jira/browse/SPARK-58633?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel
]
Ruifeng Zheng resolved SPARK-58633.
-----------------------------------
Fix Version/s: 4.4.0
Resolution: Fixed
Issue resolved by pull request 57838
[https://github.com/apache/spark/pull/57838]
> Use integer label indexes in StringIndexerModel
> -----------------------------------------------
>
> Key: SPARK-58633
> URL: https://issues.apache.org/jira/browse/SPARK-58633
> Project: Spark
> Issue Type: Sub-task
> Components: ML
> Affects Versions: 5.0.0
> Reporter: Ruifeng Zheng
> Assignee: Ruifeng Zheng
> Priority: Major
> Labels: pull-request-available
> Fix For: 4.4.0
>
>
> StringIndexerModel currently stores label indexes as Double values in an
> OpenHashMap. Store them as Int values instead to reduce the model closure
> memory footprint, and cast indexed values to Double only when producing the
> output column.
--
This message was sent by Atlassian Jira
(v8.20.10#820010)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
