zhengruifeng created SPARK-30178: ------------------------------------ Summary: RobustScaler support bigger numFeatures Key: SPARK-30178 URL: https://issues.apache.org/jira/browse/SPARK-30178 Project: Spark Issue Type: Improvement Components: ML Affects Versions: 3.0.0 Reporter: zhengruifeng
It is a bottleneck to collect the whole Array[QuantileSummaries] from executors, since a QuantileSummaries is a large object, which maintains large arrays of size 10k({color:#93a6f5}defaultCompressThreshold{color})/50k({color:#93a6f5}defaultHeadSize{color}). So we need to compute the ranges/medians more distributedly. In Spark-Shell with default params, I processed dataset with numFeatures=69,200, and current impl fail due to OOM. -- This message was sent by Atlassian Jira (v8.3.4#803005) --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org