svd上的spark mllib内存错误(单机)

时间:2015-03-30 14:05:06

标签: scala apache-spark apache-spark-mllib

我有一个大型数据文件(大约4 GB),我正在使用单个PC上的spark进行分析。

scala> x
res29: org.apache.spark.mllib.linalg.distributed.RowMatrix = org.apache.spark.mllib.linalg.distributed.RowMatrix@5a86096a

scala> x.numRows
res27: Long = 302529

scala> x.numCols
res28: Long = 1828

当我尝试计算主要组件时,我收到内存错误:

scala> val pc: Matrix = x.computePrincipalComponents(2)

     15/03/30 14:55:22 INFO ContextCleaner: Cleaned shuffle 1
    java.lang.OutOfMemoryError: Java heap space
        at breeze.linalg.svd$.breeze$linalg$svd$$doSVD_Double(svd.scala:92)
        at breeze.linalg.svd$Svd_DM_Impl$.apply(svd.scala:39)
        at breeze.linalg.svd$Svd_DM_Impl$.apply(svd.scala:38)
        at breeze.generic.UFunc$class.apply(UFunc.scala:48)
        at breeze.linalg.svd$.apply(svd.scala:22)
        at org.apache.spark.mllib.linalg.distributed.RowMatrix.computePrincipalComponents(RowMatrix.scala:380)
        at $iwC$$iwC$$iwC$$iwC$$iwC$$iwC$$iwC$$iwC$$iwC$$iwC$$iwC$$iwC.<init>(<console>:39)

我该如何解决?

1 个答案:

答案 0 :(得分:0)

如果您碰巧拥有比Spark当前使用的RAM更多的RAM,您可以尝试使用命令行选项--driver-memory 8g来增加Java堆大小(假设&#34;本地&#34;模式在这里,其中计算由驱动程序完成)。默认值仅为512米。