|
|
@@ -37,7 +37,7 @@ object train_01_xgb_ad_20240808{
|
|
|
val func_object = param.getOrElse("func_object", "binary:logistic")
|
|
|
val func_metric = param.getOrElse("func_metric", "auc")
|
|
|
val repartition = param.getOrElse("repartition", "20").toInt
|
|
|
- val modelPath = param.getOrElse("modelPath", "/root/zhangbo/recommend-model/recommend-model-produce/models/")
|
|
|
+ val modelPath = param.getOrElse("modelPath", "/dw/recommend/model/35_ad_model/model_xgb")
|
|
|
val modelFile = param.getOrElse("modelFile", "model.tar.gz")
|
|
|
|
|
|
val loader = getClass.getClassLoader
|
|
|
@@ -99,10 +99,9 @@ object train_01_xgb_ad_20240808{
|
|
|
|
|
|
|
|
|
if(modelPath.nonEmpty && modelFile.nonEmpty){
|
|
|
- val modelPathTmp = modelPath + "/tmp"
|
|
|
- model.write.overwrite.save("file://" + modelPathTmp)
|
|
|
- val gzPath = modelPath + "/" + modelFile
|
|
|
- CompressUtil.compressDirectoryToGzip(modelPath, gzPath)
|
|
|
+ model.write.overwrite.save(modelPath)
|
|
|
+// val gzPath = modelPath + "/" + modelFile
|
|
|
+// CompressUtil.compressDirectoryToGzip(modelPath, gzPath)
|
|
|
}
|
|
|
|
|
|
if (testPath.nonEmpty){
|
zhangbo