def _ranking(self):
    return (self.r + (self.lamb * torch.sqrt(((2 * torch.log(self.t)) / self.n))))
def _ranking(self):
    return (self.r + (self.lamb * torch.sqrt(((2 * torch.log(self.t)) / self.n))))