Browse Source

generate label for mysql

罗俊辉 1 năm trước cách đây
mục cha
commit
e9f2b46528
1 tập tin đã thay đổi với 4 bổ sung4 xóa
  1. 4 4
      process_data.py

+ 4 - 4
process_data.py

@@ -269,10 +269,10 @@ class UserProcess(object):
 
 
             df.append(temp[1:])
             df.append(temp[1:])
         df = pd.DataFrame(df, columns=self.user_features)
         df = pd.DataFrame(df, columns=self.user_features)
-        df['ros_30'] = np.where(int(df['user_view_30']) != 0, df['user_return_30'] / df['user_share_30'], np.nan)
-        df['rov_30'] = np.where(int(df['user_view_30']) != 0, df['user_return_30'] / df['user_view_30'], np.nan)
-        df['ros_3'] = np.where(int(df['user_view_3']) != 0, df['user_return_3'] / df['user_share_3'], np.nan)
-        df['rov_3'] = np.where(int(df['user_view_3']) != 0, df['user_return_3'] / df['user_view_3'], np.nan)
+        df['ros_30'] = np.where(df['user_view_30'] != 0, df['user_return_30'] / df['user_share_30'], np.nan)
+        df['rov_30'] = np.where(df['user_view_30'] != 0, df['user_return_30'] / df['user_view_30'], np.nan)
+        df['ros_3'] = np.where(df['user_view_3'] != 0, df['user_return_3'] / df['user_share_3'], np.nan)
+        df['rov_3'] = np.where(df['user_view_3'] != 0, df['user_return_3'] / df['user_view_3'], np.nan)
         df.to_json(des_path, orient='records')
         df.to_json(des_path, orient='records')