Merge branch 'master' of https://github.com/sephml/Python

Author: sephml

machine_learning/mab.py

@@ -159,8 +159,7 @@ def select_arm(self) -> int:
         """
         if self.total_counts < self.k:
             return self.total_counts
-        ucb_values = self.values + \
-            np.sqrt(2 * np.log(self.total_counts) / self.counts)
+        ucb_values = self.values + np.sqrt(2 * np.log(self.total_counts) / self.counts)
         return np.argmax(ucb_values)
 
     def update(self, arm_index: int, reward: int) -> None: