Skip to content

G
gm_mab
Commit 0d0ebdaf authored by 段英荣's avatar 段英荣
Browse files
Options

调试

parent fc9e41fb
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
linucb/core/Linucb.py
View file @ 0d0ebdaf
...@@ -50,7 +50,6 @@ class LinUCB: ...@@ -50,7 +50,6 @@ class LinUCB:
tag_dict = pickle.loads(redis_linucb_tag_data_dict[tag_id]) tag_dict = pickle.loads(redis_linucb_tag_data_dict[tag_id])
Aa_list.append(tag_dict["Aa"]) Aa_list.append(tag_dict["Aa"])
theta_list.append(tag_dict["theta"]) theta_list.append(tag_dict["theta"])
print(tag_dict["theta"].shape)
xa = np.array(user_features_list) xa = np.array(user_features_list)
xaT = np.transpose(xa) xaT = np.transpose(xa)
...@@ -80,6 +79,7 @@ class LinUCB: ...@@ -80,6 +79,7 @@ class LinUCB:
np_score_list = list() np_score_list = list()
np_score_dict = dict() np_score_dict = dict()
import pdb;pdb.set_trace()
for score_index in range(0,np_array.size): for score_index in range(0,np_array.size):
score = np_array.take(score_index) score = np_array.take(score_index)
np_score_list.append(score) np_score_list.append(score)
...@@ -88,6 +88,7 @@ class LinUCB: ...@@ -88,6 +88,7 @@ class LinUCB:
else: else:
np_score_dict[score].append(score_index) np_score_dict[score].append(score_index)
import pdb;pdb.set_trace()
sorted_np_score_list = sorted(np_score_list,reverse=True) sorted_np_score_list = sorted(np_score_list,reverse=True)
for top_score in sorted_np_score_list: for top_score in sorted_np_score_list:
for top_score_index in np_score_dict[top_score]: for top_score_index in np_score_dict[top_score]:
......
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