Limit sensitivity for learning

es-meta · Feb 10, 2025 · ede0eab · ede0eab
1 parent 216da25
commit ede0eab
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/main/scala/esmeta/es/util/fuzzer/TargetFeatureSet.scala b/src/main/scala/esmeta/es/util/fuzzer/TargetFeatureSet.scala
@@ -66,7 +66,9 @@ case class TargetFeatureSet(
 
   def touchWithHit(stacks: Iterable[List[String]]): Unit =
     stacks.foreach {
-      _.foreach { feature =>
+      _.take(
+        config.maxSensitivity,
+      ).foreach { feature =>
         val tmpData = targetFeatureMap.getOrElse(feature, TargetFeatureData())
         tmpData.hit(
           rootHits,
@@ -80,7 +82,9 @@ case class TargetFeatureSet(
 
   def touchWithMiss(stacks: Iterable[List[String]]): Unit =
     stacks.foreach {
-      _.foreach { feature =>
+      _.take(
+        config.maxSensitivity,
+      ).foreach { feature =>
         val tmpData = targetFeatureMap.getOrElse(feature, TargetFeatureData())
         tmpData.miss(
           rootHits,