modify

1dad0805 · 段英荣 · 00a2be1e · 1dad0805
Commit 1dad0805 authored Mar 22, 2019 by 段英荣
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

linucb.py linucb/views/linucb.py +2 -0

No files found.
--- a/linucb/views/linucb.py
+++ b/linucb/views/linucb.py
@@ -86,6 +86,8 @@ class LinUCB:
            for top_score in sorted_np_score_list:
                for top_score_index in np_score_dict[top_score]:
                    top_tag_set.add(str(tag_list[top_score_index], encoding="utf-8"))
+                    if len(top_tag_set) >= 10:
+                        break

                if len(top_tag_set) >= 10:
                    break