Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
483a16d
1
Parent(s): 676b027
Update make_dataset.py
Browse files- src/data/make_dataset.py +1 -0
src/data/make_dataset.py
CHANGED
|
@@ -8,6 +8,7 @@ import pprint
|
|
| 8 |
def make_dataset(dataset="cnn_dailymail", split="train"):
|
| 9 |
"""make dataset for summarisation"""
|
| 10 |
if not os.path.exists("data/raw"):
|
|
|
|
| 11 |
df = pd.DataFrame()
|
| 12 |
df['article'] = dataset['article']
|
| 13 |
df['highlights'] = dataset['highlights']
|
|
|
|
| 8 |
def make_dataset(dataset="cnn_dailymail", split="train"):
|
| 9 |
"""make dataset for summarisation"""
|
| 10 |
if not os.path.exists("data/raw"):
|
| 11 |
+
os.makedirs("data/raw")
|
| 12 |
df = pd.DataFrame()
|
| 13 |
df['article'] = dataset['article']
|
| 14 |
df['highlights'] = dataset['highlights']
|