|
@@ -38,6 +38,7 @@ object makedata_02_writeredis {
|
|
|
val savePathUser = param.getOrDefault("savePathUser", "")
|
|
|
val savePathVideo = param.getOrDefault("savePathVideo", "")
|
|
|
val userSampleIDs = param.getOrDefault("userSampleIDs", "")
|
|
|
+ val sampleRate = param.getOrDefault("sampleRate", "1.0").toDouble
|
|
|
// val userSampleIDsPathFix = param.getOrDefault("userSampleIDsPathFix", "")
|
|
|
// /dw/recommend/model/feature/
|
|
|
|
|
@@ -127,7 +128,9 @@ object makedata_02_writeredis {
|
|
|
if (userSampleIDs.nonEmpty) {
|
|
|
savePathPart = savePathPart + "_" + userSampleIDs
|
|
|
}
|
|
|
- val userDataRead = sc.textFile(savePathPart).filter(_.split("\t").length >= 2).map(r => {
|
|
|
+ val userDataRead = sc.textFile(savePathPart).filter(_.split("\t").length >= 2)
|
|
|
+ .sample(false, sampleRate)
|
|
|
+ .map(r => {
|
|
|
val rList = r.split("\t")
|
|
|
(rList(0), rList(1))
|
|
|
})
|