调试

3f869f3b · 段英荣 · ac8122d6 · 3f869f3b
Commit 3f869f3b authored Apr 20, 2020 by 段英荣
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

Linucb.py linucb/core/Linucb.py +2 -2

No files found.
--- a/linucb/core/Linucb.py
+++ b/linucb/core/Linucb.py
@@ -94,10 +94,10 @@ class LinUCB:
                    tag_id = tag_list[top_score_index]
                    top_tag_dict[tag_id] = top_score
                    top_tag_set.add(tag_id)
-                    if len(top_tag_dict) >= 20:
+                    if len(top_tag_dict) >= 5:
                        break

-                if len(top_tag_dict) >= 20:
+                if len(top_tag_dict) >= 5:
                    break

            return (top_tag_dict,top_tag_set)