Gagan Bhatia commited on
Commit
1b154c5
1 Parent(s): 563a322

Update process_data.py

Browse files
Files changed (1) hide show
  1. src/data/process_data.py +1 -0
src/data/process_data.py CHANGED
@@ -2,6 +2,7 @@ import pandas as pd
2
 
3
 
4
  def process_data(split='train'):
 
5
  df.columns = ['Unnamed: 0', 'input_text', 'output_text']
6
  print(df.columns)
7
  df.to_csv('C:/Users/gbhat/Documents/GitHub/summarization/data/processed/{}.csv'.format(split))
 
2
 
3
 
4
  def process_data(split='train'):
5
+
6
  df.columns = ['Unnamed: 0', 'input_text', 'output_text']
7
  print(df.columns)
8
  df.to_csv('C:/Users/gbhat/Documents/GitHub/summarization/data/processed/{}.csv'.format(split))