fix(soak): scale also rarity

[csit.git] / resources / libraries / python / PLRsearch / Integrator.py

diff --git a/resources/libraries/python/PLRsearch/Integrator.py b/resources/libraries/python/PLRsearch/Integrator.py
index a7a5939..7f118db 100644 (file)
--- a/resources/libraries/python/PLRsearch/Integrator.py
+++ b/resources/libraries/python/PLRsearch/Integrator.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2021 Cisco and/or its affiliates.
+# Copyright (c) 2022 Cisco and/or its affiliates.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at:
@@ -253,7 +253,7 @@ def estimate_nd(communication_pipe, scale_coeff=8.0, trace_enabled=False):
         )
         # The code above looked at weight (not importance).
         # The code below looks at importance (not weight).
-        log_rarity = param_distance / 2.0
+        log_rarity = param_distance / 2.0 / scale_coeff
         trace(u"log_rarity", log_rarity)
         log_importance = log_weight + log_rarity
         trace(u"log_importance", log_importance)