dssm train

recommend-model-produce/src/main/python/models/dssm/milvus_import.py

@@ -18,7 +18,7 @@ configs = {
     "hadoop.job.ugi": ""
 }
 hdfs_client = MyHDFSClient(hadoop_home, configs)
-oss_client = HangZhouOSSClient("art-recommend")
+oss_client = HangZhouOSSClient("tz-milvus")
 
 
 def process_file(file_path):
@@ -30,7 +30,7 @@ def process_file(file_path):
         if(len(sample_values) == 2):
             vid, vec = sample_values
             result.append({
-                "vid":vid,
+                "vid":int(vid),
                 "vec":json.loads(vec)
             })
         else :