modify linucb

2c6d0800 · 段英荣 · ed2ec085 · 2c6d0800
Commit 2c6d0800 authored Feb 28, 2019 by 段英荣
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

linucb.py linucb/views/linucb.py +1 -1

No files found.
--- a/linucb/views/linucb.py
+++ b/linucb/views/linucb.py
@@ -61,7 +61,7 @@ class LinUCB:

            art_max = tag_list[np.argmax(np.dot(xaT, theta_tmp) + cls.alpha * np.sqrt(np.dot(np.dot(xaT, AaI_tmp), xa)))]

-            return [art_max]
+            return [int(art_max)]
        except:
            logging.error("catch exception,err_msg:%s" % traceback.format_exc())
            return []