service model 优化

61864bf7 · 郭羽 · 330049e1 · 61864bf7
Commit 61864bf7 authored Jul 30, 2021 by 郭羽
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

featureEng.py spark/featureEng.py +1 -0

No files found.
--- a/spark/featureEng.py
+++ b/spark/featureEng.py
@@ -219,6 +219,7 @@ def arrayReverse(arr):

 def addUserFeatures(samples,dataVocab,multiVocab):
    dataVocab["userid"] = collectColumnToVocab(samples, "userid")
+    dataVocab["user_city_id"] = collectColumnToVocab(samples, "user_city_id")
    dataVocab["user_os"] = ["ios","android"]

    extractTagsUdf = F.udf(extractTags, ArrayType(StringType()))