Ver Fonte

83 sample

jch há 1 mês atrás
pai
commit
9aeffcf6c4

+ 32 - 1
src/main/scala/com/aliyun/odps/spark/examples/makedata_recsys_r_rate/makedata_recsys_82_nor_sample_20250221.scala

@@ -12,6 +12,33 @@ import scala.io.Source
 import scala.util.Random
 
 object makedata_recsys_82_nor_sample_20250221 {
+  val hoursMap = Map(
+    "hour@1" -> 1.0,
+    "hour@2" -> 2.0,
+    "hour@3" -> 3.0,
+    "hour@4" -> 4.0,
+    "hour@5" -> 5.0,
+    "hour@6" -> 6.0,
+    "hour@7" -> 7.0,
+    "hour@8" -> 8.0,
+    "hour@9" -> 9.0,
+    "hour@10" -> 10.0,
+    "hour@11" -> 11.0,
+    "hour@12" -> 12.0,
+    "hour@13" -> 13.0,
+    "hour@14" -> 14.0,
+    "hour@15" -> 15.0,
+    "hour@16" -> 16.0,
+    "hour@17" -> 17.0,
+    "hour@18" -> 18.0,
+    "hour@19" -> 19.0,
+    "hour@20" -> 20.0,
+    "hour@21" -> 21.0,
+    "hour@22" -> 22.0,
+    "hour@23" -> 23.0,
+    "hour@24" -> 24.0
+  )
+
   def main(args: Array[String]): Unit = {
     // 1 读取参数
     val param = ParamUtils.parseArgs(args)
@@ -162,7 +189,11 @@ object makedata_recsys_82_nor_sample_20250221 {
     if (data.nonEmpty) {
       val obj = JSON.parseObject(data)
       obj.foreach(r => {
-        features.put(r._1, obj.getDoubleValue(r._1))
+        if (hoursMap.contains(r._1)) {
+          features.put("hour", hoursMap(r._1))
+        } else {
+          features.put(r._1, obj.getDoubleValue(r._1))
+        }
       })
     }
     features