def __init__(self, n_arms):
    super(ucb_user, self).__init__()
    self.n_arms = n_arms
    self.t = torch.tensor(1.0)
    self.tau
    self.sw_r = []
    self.sw_arms = []
    self.n = torch.zeros(self.n_arms)
    self.r = torch.zeros(self.n_arms)
    self.alpha = 0.9
    self.lamb = 1
    self.id = (- 1)
    self.returns = 0
def __init__(self, n_arms):
    super(ucb_user, self).__init__()
    self.n_arms = n_arms
    self.t = torch.tensor(1.0)
    self.tau
    self.sw_r = []
    self.sw_arms = []
    self.n = torch.zeros(self.n_arms)
    self.r = torch.zeros(self.n_arms)
    self.alpha = 0.9
    self.lamb = 1
    self.id = (- 1)
    self.returns = 0