init

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

@@ -102,14 +102,14 @@ public class XGBoostPredict {
             String objectName = "test/model.tar.gz";
             OSSService ossService = new OSSService();
 
-            String destPath = "file:///root/recommend-model/model2.tar.gz";
+            String destPath = "/root/recommend-model/model2.tar.gz";
             ossService.download(bucketName, destPath, objectName);
-            String destDir = "file:///root/recommend-model/modelpredict";
+            String destDir = "/root/recommend-model/modelpredict";
             CompressUtil.decompressGzFile(destPath, destDir);
 
             // 显示预测结果
 
-            XGBoostClassificationModel model2 = XGBoostClassificationModel.load(destDir);
+            XGBoostClassificationModel model2 = XGBoostClassificationModel.load("file://" + destDir);
             Dataset<Row> predictions = model2.transform(assembledData);
             predictions.select("label", "prediction", "features", "rawPrediction", "probability").show(500);
 

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

@@ -116,8 +116,8 @@ public class XGBoostTrain {
             // 保存模型
             String path = "file:///root/recommend-model/modeltrain";
             model.write().overwrite().save(path);
-            String outputPath = "file:///root/recommend-model/model.tar.gz";
-            CompressUtil.compressDirectoryToGzip(path, outputPath);
+            String outputPath = "/root/recommend-model/model.tar.gz";
+            CompressUtil.compressDirectoryToGzip(path, "file://" + outputPath);
             String bucketName = "art-test-video";
             String ossPath = "test/model.tar.gz";
             OSSService ossService = new OSSService();