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']
|