Skip to content

Commit d942017

Browse files
committed
rename to setIntermediateRDDStorageLevel
1 parent 7550029 commit d942017

File tree

1 file changed

+5
-5
lines changed
  • mllib/src/main/scala/org/apache/spark/mllib/recommendation

1 file changed

+5
-5
lines changed

mllib/src/main/scala/org/apache/spark/mllib/recommendation/ALS.scala

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ class ALS private (
115115
private var nonnegative = false
116116

117117
/** storage level for user/product in/out links */
118-
private var intermediateDataStorageLevel: StorageLevel = StorageLevel.MEMORY_AND_DISK
118+
private var intermediateRDDStorageLevel: StorageLevel = StorageLevel.MEMORY_AND_DISK
119119

120120
/**
121121
* Set the number of blocks for both user blocks and product blocks to parallelize the computation
@@ -199,8 +199,8 @@ class ALS private (
199199
* set `spark.rdd.compress` to `true` to reduce the space requirement, at the cost of speed.
200200
*/
201201
@DeveloperApi
202-
def setIntermediateDataStorageLevel(storageLevel: StorageLevel): this.type = {
203-
this.intermediateDataStorageLevel = storageLevel
202+
def setIntermediateRDDStorageLevel(storageLevel: StorageLevel): this.type = {
203+
this.intermediateRDDStorageLevel = storageLevel
204204
this
205205
}
206206

@@ -456,8 +456,8 @@ class ALS private (
456456
}, preservesPartitioning = true)
457457
val inLinks = links.mapValues(_._1)
458458
val outLinks = links.mapValues(_._2)
459-
inLinks.persist(intermediateDataStorageLevel)
460-
outLinks.persist(intermediateDataStorageLevel)
459+
inLinks.persist(intermediateRDDStorageLevel)
460+
outLinks.persist(intermediateRDDStorageLevel)
461461
(inLinks, outLinks)
462462
}
463463

0 commit comments

Comments
 (0)