丁云鹏 hace 8 meses
padre
commit
2ccfe4da0f

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

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