Skip to content

F
ffm-baseline
Commit fff6fec0 authored by 高雅喆's avatar 高雅喆
Browse files
Options

update

parent fd7d5a83
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 5 deletions
eda/smart_rank/gyz_test.py
View file @ fff6fec0
......@@ -97,11 +97,8 @@ spark.sparkContext.setLogLevel("WARN")
spark.sparkContext.addPyFile("/srv/apps/ffm-baseline_git/eda/smart_rank/tool.py")
device_ids_lst_rdd = spark.sparkContext.parallelize(device_info)
result = device_ids_lst_rdd.repartition(100).map(lambda x: get_user_service_portrait(x, all_word_tags, all_tag_tag_type, all_3tag_2tag, all_tags_name, size=None, pay_time=pay_time))
print(result.take(1))
result1 = result.map(
lambda x: (str(x[0]), str(x[1]), str(x[2]))
)
df = spark.createDataFrame(result1).toDF("device", "search_words", "user_portrait").toPandas()
print(result.take(10))
df = spark.createDataFrame(result).toDF("device", "search_words", "user_portrait").toPandas()
df.to_csv("~/test_df.csv", index=False)
......
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