Gagan Bhatia commited on
Commit
66ca39d
1 Parent(s): 1528dbf

Update process_data.py

Browse files
Files changed (1) hide show
  1. src/data/process_data.py +1 -2
src/data/process_data.py CHANGED
@@ -5,8 +5,7 @@ def process_data(split='train'):
5
 
6
  df = pd.read_csv('data/raw/{}.csv'.format(split))
7
  df.columns = ['Unnamed: 0', 'input_text', 'output_text']
8
- print(df.columns)
9
- df.to_csv('C:/Users/gbhat/Documents/GitHub/summarization/data/processed/{}.csv'.format(split))
10
 
11
 
12
  if __name__ == '__main__':
 
5
 
6
  df = pd.read_csv('data/raw/{}.csv'.format(split))
7
  df.columns = ['Unnamed: 0', 'input_text', 'output_text']
8
+ df.to_csv('data/processed/{}.csv'.format(split))
 
9
 
10
 
11
  if __name__ == '__main__':