Skip to content

P
physical
Commit ae9414e2 authored by 段英荣's avatar 段英荣
Browse files
Options

modify top_tag_list_len

parent f3d6ae58
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
linucb/views/linucb.py
View file @ ae9414e2
...@@ -59,7 +59,7 @@ class LinUCB: ...@@ -59,7 +59,7 @@ class LinUCB:
AaI_tmp = np.array(Aa_list) AaI_tmp = np.array(Aa_list)
theta_tmp = np.array(theta_list) theta_tmp = np.array(theta_list)
top_tag_list_len = len(tag_list)/3 top_tag_list_len = int(len(tag_list)/3)
top_np_ind = np.argpartition(np.dot(xaT, theta_tmp) + cls.alpha * np.sqrt(np.dot(np.dot(xaT, AaI_tmp), xa)), -top_tag_list_len)[-top_tag_list_len:] top_np_ind = np.argpartition(np.dot(xaT, theta_tmp) + cls.alpha * np.sqrt(np.dot(np.dot(xaT, AaI_tmp), xa)), -top_tag_list_len)[-top_tag_list_len:]
top_tag_list = list() top_tag_list = list()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment