ソースを参照

video keywords

jch 1 ヶ月 前
コミット
68566efb12

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

@@ -283,6 +283,17 @@ public class FeatureTransform {
                 featMap.put(key, 1.0);
             }
         }
+        if (videoInfo.containsKey("keywords")) {
+            String keywords = videoInfo.get("keywords");
+            if (null != keywords && !keywords.isEmpty()) {
+                for (String kw : keywords.split(",")) {
+                    if (!kw.isEmpty()) {
+                        String featKey = String.format("%s@kw@%s", prefix, kw);
+                        featMap.put(featKey, 1.0);
+                    }
+                }
+            }
+        }
     }
 
     private static void getTwoVideoCrossFeature(String prefix, List<String> attrs, Map<String, String> video1, Map<String, String> video2, Map<String, Double> featMap) {