Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
06d3c94
1
Parent(s):
e0a340d
Update make_dataset.py
Browse files- src/data/make_dataset.py +1 -0
src/data/make_dataset.py
CHANGED
|
@@ -7,6 +7,7 @@ def make_dataset(dataset='cnn_dailymail', split='train'):
|
|
| 7 |
dataset = load_dataset(dataset, '3.0.0', split=split)
|
| 8 |
df = pd.DataFrame()
|
| 9 |
df['article'] = dataset['article']
|
|
|
|
| 10 |
|
| 11 |
if __name__ == '__main__':
|
| 12 |
make_dataset(dataset='cnn_dailymail', split='train', version="3.0.0")
|
|
|
|
| 7 |
dataset = load_dataset(dataset, '3.0.0', split=split)
|
| 8 |
df = pd.DataFrame()
|
| 9 |
df['article'] = dataset['article']
|
| 10 |
+
df['highlights'] = dataset['highlights']
|
| 11 |
|
| 12 |
if __name__ == '__main__':
|
| 13 |
make_dataset(dataset='cnn_dailymail', split='train', version="3.0.0")
|