Browse Source

generate label for mysql

罗俊辉 1 year ago
parent
commit
40838fa904
1 changed files with 1 additions and 1 deletions
  1. 1 1
      process_data.py

+ 1 - 1
process_data.py

@@ -157,7 +157,7 @@ class SpiderProcess(object):
                 continue
         df = pd.DataFrame(df, columns=['title', 'channel', 'out_user_id', 'mode', 'out_play_cnt', 'out_like_cnt',
                                        'out_share_cnt', 'lop', 'duration', 'tag1', 'tag2', 'tag3'])
-        df.to_json("data/train_data/spider_data_{}.json".format(datetime.datetime.today().strftime("y%m%d")))
+        df.to_json("data/train_data/spider_data_{}.json".format(datetime.datetime.today().strftime("y%m%d")), orient='records')
 
 
 class UserProcess(object):