Commit 3855baf5 authored by 郭羽's avatar 郭羽

service model 优化

parent 4875b3b8
......@@ -148,6 +148,8 @@ def addItemFeatures(itemDF,dataVocab,multi_col_vocab):
for col in onehot_col:
new_c = ITEM_PREFIX + col
dataVocab[new_c] = list(set(itemDF[col].tolist()))
itemDF[new_c] = itemDF[col]
itemDF = itemDF.drop(columns=[onehot_col])
for c in multi_col:
multi_col_vocab[c] = list(set(itemDF[c].tolist()))
......
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