Bladeren bron

add metric

jch 2 maanden geleden
bovenliggende
commit
920335b98a

+ 8 - 0
recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/MetricUtils.scala

@@ -20,6 +20,14 @@ object MetricUtils {
     }
   }
 
+  def clipLabel(label: Double, maxVal: Double = 30): Double = {
+    if (label < maxVal) {
+      label
+    } else {
+      maxVal + 2 * Math.log(label - maxVal + 1)
+    }
+  }
+
   def calMAPE(evalRdd: RDD[Row]): Double = {
     val apeRdd = evalRdd.map(raw => {
       val label = raw.get(0).toString.toDouble