From e6034bd188076d69e09fcc111138679feb84e826 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Fri, 4 Apr 2025 15:34:01 +0200 Subject: [PATCH] refactored --- k-bandits.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/k-bandits.py b/k-bandits.py index f3f838d..433d0e3 100644 --- a/k-bandits.py +++ b/k-bandits.py @@ -20,16 +20,13 @@ def simple_max(Q, N, t, _tie_break): am = np.argmax(Q + _tie_break[t]) return am - def ucb(Q, N, t, _tie_break): c = 2 if N.min() == 0: -# return np.argmax(N + _tie_break[t]) return np.random.choice(np.flatnonzero(N == N.min())) M = Q + c * np.sqrt(np.divide(np.log(t), N)) return np.argmax(M + _tie_break[t]) # breaking ties randomly -# return np.argmax(M) # breaking ties randomly class Param: def __init__(self, epsilon:float=0, rho:float=0, alpha:float =0, q_ic:float=0, argmax_func=simple_max):