Skip to content

S
serviceRec
Commit 3855baf5 authored by 郭羽's avatar 郭羽
Browse files
Options

service model 优化

parent 4875b3b8
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
spark/featureEng2.py
View file @ 3855baf5
...@@ -148,6 +148,8 @@ def addItemFeatures(itemDF,dataVocab,multi_col_vocab): ...@@ -148,6 +148,8 @@ def addItemFeatures(itemDF,dataVocab,multi_col_vocab):
for col in onehot_col: for col in onehot_col:
new_c = ITEM_PREFIX + col new_c = ITEM_PREFIX + col
dataVocab[new_c] = list(set(itemDF[col].tolist())) dataVocab[new_c] = list(set(itemDF[col].tolist()))
itemDF[new_c] = itemDF[col]
itemDF = itemDF.drop(columns=[onehot_col])
for c in multi_col: for c in multi_col:
multi_col_vocab[c] = list(set(itemDF[c].tolist())) 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