Skip to content

Commit 9e07925

Browse files
committed
Update create_csv.py
1 parent 78452fb commit 9e07925

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

create_csv.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -100,15 +100,15 @@ def write_custom_csv(emotions=['sad', 'neutral', 'happy'], train_name="train_cus
100100
for i, file in enumerate(glob.glob(f"data/train-custom/*_{category}.wav")):
101101
train_target["path"].append(file)
102102
train_target["emotion"].append(category)
103-
if verbose:
104-
print(f"[Custom Dataset] There are {i} training audio files for category:{category}")
103+
# if verbose:
104+
# print(f"[Custom Dataset] There are {i} training audio files for category:{category}")
105105

106106
# test data
107107
for i, file in enumerate(glob.glob(f"data/test-custom/*_{category}.wav")):
108108
test_target["path"].append(file)
109109
test_target["emotion"].append(category)
110-
if verbose:
111-
print(f"[Custom Dataset] There are {i} testing audio files for category:{category}")
110+
# if verbose:
111+
# print(f"[Custom Dataset] There are {i} testing audio files for category:{category}")
112112

113113
if train_target["path"]:
114114
pd.DataFrame(train_target).to_csv(train_name)

0 commit comments

Comments
 (0)