add data

98479180 · 张彦钊 · e384f10f · 98479180
Commit 98479180 authored Dec 11, 2018 by 张彦钊
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

ffm.py tensnsorflow/ffm.py +1 -3

No files found.
--- a/tensnsorflow/ffm.py
+++ b/tensnsorflow/ffm.py
@@ -48,9 +48,7 @@ def get_data():
    print("shape")
    print(df.shape)
    device_tuple = tuple(set(df["device_id"].values.tolist()))
-    a = device_tuple[0:2]
-    print(a)
-    df = pd.merge(df,get_statistics(a),on="device_id",how='left').fillna(0)
+    df = pd.merge(df,get_statistics(device_tuple),on="device_id",how='left').fillna(0)
    print("merge")
    print(df.head())
    print("shape")