feat:添加模型保存逻辑

recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/recsys_01_xgb_train.scala

@@ -37,6 +37,8 @@ object recsys_01_xgb_train {
     val func_metric = param.getOrElse("func_metric", "auc")
     val repartition = param.getOrElse("repartition", "20").toInt
     val subsample = param.getOrElse("subsample", "0.95").toDouble
+    val modelPath = param.getOrElse("modelPath", "/dw/recommend/model/45_recommend_model/")
+    val modelFile = param.getOrElse("modelFile", "model.tar.gz")
 
     val loader = getClass.getClassLoader
     val resourceUrl = loader.getResource(featureFile)
@@ -97,6 +99,9 @@ object recsys_01_xgb_train {
       .setMinChildWeight(1)
     val model = xgbClassifier.fit(xgbInput)
 
+    if (modelPath.nonEmpty && modelFile.nonEmpty) {
+      model.write.overwrite().save(modelPath)
+    }
 
     val testData = createData4Ad(
       sc.textFile(testPath),