jch il y a 2 mois
Parent
commit
efbfcb5b35

+ 9 - 9
src/main/scala/com/aliyun/odps/spark/examples/myUtils/FeatureTransform.java

@@ -189,8 +189,8 @@ public class FeatureTransform {
                         }
                         if (null != hVideoMap && hVideoMap.containsKey(vid)) {
                             Map<String, String> hVideo = hVideoMap.get(vid);
-                            getVideoCateFeature(prefix, hVideo, featMap);
-                            getTwoVideoCrossFeature(prefix, hVideoSimAttrs, hVideo, rankVideo, featMap);
+                            getVideoCateFeature(baseKey, hVideo, featMap);
+                            getTwoVideoCrossFeature(baseKey, hVideoSimAttrs, hVideo, rankVideo, featMap);
                         }
                     }
                 }
@@ -328,13 +328,13 @@ public class FeatureTransform {
                 double ros_minus = getOneInfo(calType + "_ros_minus_" + period, infoMap);
                 double rovn = getOneInfo(calType + "_rovn_" + period, infoMap);
 
-                featMap.put(prefix + "_" + period + "_" + calType + "_str_one_", str_one);
-                featMap.put(prefix + "_" + period + "_" + calType + "_ros_one_", ros_one);
-                featMap.put(prefix + "_" + period + "_" + calType + "_str_", str);
-                featMap.put(prefix + "_" + period + "_" + calType + "_ros_", ros);
-                featMap.put(prefix + "_" + period + "_" + calType + "_str_plus_", str_plus);
-                featMap.put(prefix + "_" + period + "_" + calType + "_ros_minus_", ros_minus);
-                featMap.put(prefix + "_" + period + "_" + calType + "_rovn_", rovn);
+                featMap.put(prefix + "_" + period + "_" + calType + "_str_one", str_one);
+                featMap.put(prefix + "_" + period + "_" + calType + "_ros_one", ros_one);
+                featMap.put(prefix + "_" + period + "_" + calType + "_str", str);
+                featMap.put(prefix + "_" + period + "_" + calType + "_ros", ros);
+                featMap.put(prefix + "_" + period + "_" + calType + "_str_plus", str_plus);
+                featMap.put(prefix + "_" + period + "_" + calType + "_ros_minus", ros_minus);
+                featMap.put(prefix + "_" + period + "_" + calType + "_rovn", rovn);
             }
         }
     }