diff --git a/tensnsorflow/multi.py b/tensnsorflow/multi.py index f7292a91956dcf60817092871694fbe9f064ee36..51a28dd400c492e7e187d43d40a626a4925236d5 100644 --- a/tensnsorflow/multi.py +++ b/tensnsorflow/multi.py @@ -92,7 +92,7 @@ def get_predict(date,value_map,app_list_map): df = df.na.fill(dict(zip(features, features))) rdd = df.select("level2_ids","ucity_id","device_id","cid_id","label", "y", "z") \ .rdd.map(lambda x: (app_list_func(x[0], app_list_map),x[1],x[2],x[3],x[4],float(x[5]),float(x[6]), - [value_map.get(x[1], 300000),value_map.get(date, 299999)])) + [value_map.get(x[1], 299999),value_map.get(date, 299998)])) rdd.persist()