refactor(qav3): add Boltzmann operator

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>

diff --git a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
index 8c71392a9feb66fd4a0c983e0491a34ae98119f1..a2527ce3ca00edc1c43f8099cb34a19cde5a05e9 100644 (file)
--- a/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
+++ b/quickadapter/user_data/freqaimodels/QuickAdapterRegressorV3.py
@@ -999,6 +999,22 @@ def smoothed_min(series: pd.Series, temperature=1.0) -> float:
     return -sp.special.logsumexp(-temperature * data_array) / temperature
 
 
+def boltzmann_operator(series: pd.Series, alpha: float) -> float:
+    """
+    Compute the Boltzmann operator of an series with parameter alpha.
+    """
+    data_array = series.to_numpy()
+    if data_array.size == 0:
+        return np.nan
+    if alpha == 0:
+        return np.mean(data_array)
+    scaled_data = alpha * data_array
+    shifted_exponentials = np.exp(scaled_data - np.max(scaled_data))
+    numerator = np.sum(data_array * shifted_exponentials)
+    denominator = np.sum(shifted_exponentials)
+    return numerator / denominator
+
+
 def round_to_nearest(value: float, step: int) -> int:
     """
     Round a value to the nearest multiple of a given step.