丁云鹏 8 ماه پیش
والد
کامیت
e2918aff8f

+ 2 - 2
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/xgboost/XGBoostPredict.java

@@ -102,9 +102,9 @@ public class XGBoostPredict {
             String objectName = "test/model.tar.gz";
             OSSService ossService = new OSSService();
 
-            String destPath = "/root/recommend-model/model2.tar.gz";
+            String destPath = "file:///root/recommend-model/model2.tar.gz";
             ossService.download(bucketName, destPath, objectName);
-            String destDir = "/root/recommend-model/modelpredict";
+            String destDir = "file:///root/recommend-model/modelpredict";
             CompressUtil.decompressGzFile(destPath, destDir);
 
             // 显示预测结果

+ 2 - 2
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/xgboost/XGBoostTrain.java

@@ -114,9 +114,9 @@ public class XGBoostTrain {
             XGBoostClassificationModel model = xgbClassifier.fit(assembledData);
 
             // 保存模型
-            String path = "/root/recommend-model/modeltrain";
+            String path = "file:///root/recommend-model/modeltrain";
             model.write().overwrite().save(path);
-            String outputPath = "/root/recommend-model/model.tar.gz";
+            String outputPath = "file:///root/recommend-model/model.tar.gz";
             CompressUtil.compressDirectoryToGzip(path, outputPath);
             String bucketName = "art-test-video";
             String ossPath = "test/model.tar.gz";