调试

tools/generate_content_user_feature.py

@@ -139,7 +139,7 @@ class Generate_Feature_Info(object):
                         if len(offi_tags_v3)>0:
                             try:
                                 tag_label = label_encoder.transform(offi_tags_v3)[0]
-                                diary_feature_list = onehot_encoder.transform([tag_label]).toarray().tolist()
+                                diary_feature_list = onehot_encoder.transform([[tag_label]]).toarray().tolist()
                                 redis_client.hset(redis_name_content_linucb_feature, diary_id,
                                               json.dumps(diary_feature_list))
                             except: