Skip to content

S
serviceRec
Commit 61864bf7 authored by 郭羽's avatar 郭羽
Browse files
Options

service model 优化

parent 330049e1
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
spark/featureEng.py
View file @ 61864bf7
...@@ -219,6 +219,7 @@ def arrayReverse(arr): ...@@ -219,6 +219,7 @@ def arrayReverse(arr):
def addUserFeatures(samples,dataVocab,multiVocab): def addUserFeatures(samples,dataVocab,multiVocab):
dataVocab["userid"] = collectColumnToVocab(samples, "userid") dataVocab["userid"] = collectColumnToVocab(samples, "userid")
dataVocab["user_city_id"] = collectColumnToVocab(samples, "user_city_id")
dataVocab["user_os"] = ["ios","android"] dataVocab["user_os"] = ["ios","android"]
extractTagsUdf = F.udf(extractTags, ArrayType(StringType())) extractTagsUdf = F.udf(extractTags, ArrayType(StringType()))
......
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