jch преди 2 месеца
родител
ревизия
920335b98a
променени са 1 файла, в които са добавени 8 реда и са изтрити 0 реда
  1. 8 0
      recommend-model-produce/src/main/scala/com/tzld/piaoquan/recommend/model/MetricUtils.scala

+ 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