From a560372b5cad004418f9bea9a40731b0f6a59ac1 Mon Sep 17 00:00:00 2001
From: ZhuangYumin <zhuangyumin@sjtu.edu.cn>
Date: Thu, 28 Sep 2023 11:38:24 +0800
Subject: [PATCH] make average adjustable

---
 src/include/client.h | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/include/client.h b/src/include/client.h
index b23405b..791c9b4 100644
--- a/src/include/client.h
+++ b/src/include/client.h
@@ -425,8 +425,9 @@ inline double GetProb(double default_p,const std::vector<double> &ps)
 {
   if(ps.empty()) return default_p;
   double res=0;
-  for(int i=0;i<ps.size();i++) res+=ps[i];
-  return res/ps.size();
+  const double v=2;
+  for(int i=0;i<ps.size();i++) res+=pow(ps[i],v);
+  return pow(res/ps.size(),1.0/v);
 }
 /**
  * @brief The definition of function SimpleGuess()