瀏覽代碼

dssm train

丁云鹏 4 月之前
父節點
當前提交
7f0492c020

+ 2 - 1
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/i2i/I2IDSSMPredict.java

@@ -38,6 +38,7 @@ public class I2IDSSMPredict {
 
         String gzPath2 = "/root/recommend-model/model.tar.gz";
         ossService2.download(bucketName2, gzPath2, objectName2);
+
         hdfsService.copyFromLocalFile(gzPath2, "/dyp/dssm/model.tar.gz");
         // 加载模型
         SparkSession spark = SparkSession.builder()
@@ -56,7 +57,7 @@ public class I2IDSSMPredict {
 //
 //            String gzPath = "/root/recommend-model/model.tar.gz";
 //            ossService.download(bucketName, gzPath, objectName);
-            String modelDir = "/root/recommend-model";
+            // String modelDir = "/root/recommend-model";
             hdfsService.copyToLocalFile("/dyp/dssm/model.tar.gz", "./model.tar.gz");
             CompressUtil.decompressGzFile("./model.tar.gz", ".");
 

+ 2 - 2
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/service/OSSService.java

@@ -17,8 +17,8 @@ import java.io.Serializable;
 public class OSSService implements Serializable {
     private String accessId = "LTAI5tHMkNaRhpiDB1yWMZPn";
     private String accessKey = "XLi5YUJusVwbbQOaGeGsaRJ1Qyzbui";
-    //private String endpoint = "https://oss-cn-hangzhou-internal.aliyuncs.com";
-    private String endpoint = "https://oss-cn-hangzhou.aliyuncs.com";
+    private String endpoint = "https://oss-cn-hangzhou-internal.aliyuncs.com";
+    //private String endpoint = "https://oss-cn-hangzhou.aliyuncs.com";
 
     public void upload(String bucketName, String localFile, String objectName) {
         OSS ossClient = new OSSClientBuilder().build(endpoint, accessId, accessKey);