Skip to content

Commit fbeb0f4

Browse files
committed
Update latif2019 experiment script.
1 parent 5c0203d commit fbeb0f4

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

scripts/training/latif2019.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,16 +11,14 @@
1111
import tensorflow as tf
1212
from emotion_recognition.classification import (print_results,
1313
within_corpus_cross_validation)
14-
from emotion_recognition.dataset import RawDataset
14+
from emotion_recognition.dataset import LabelledDataset
1515
from emotion_recognition.tensorflow.classification import TFClassifier
1616
from emotion_recognition.tensorflow.models import latif2019_model
1717
from emotion_recognition.tensorflow.utils import create_tf_dataset_ragged
1818
from sklearn.model_selection import LeaveOneGroupOut
1919
from tensorflow.keras.callbacks import EarlyStopping, ReduceLROnPlateau
2020
from tensorflow.keras.optimizers import RMSprop
2121

22-
RESULTS_DIR = 'results/latif2019'
23-
2422

2523
def get_tf_dataset(x: np.ndarray, y: np.ndarray, shuffle: bool = True,
2624
batch_size: int = 16):
@@ -54,8 +52,8 @@ def main():
5452
tf.keras.backend.clear_session()
5553

5654
for corpus in ['iemocap', 'msp-improv']:
57-
dataset = RawDataset('datasets/{}/files.txt'.format(corpus),
58-
corpus=corpus)
55+
# dataset = LabelledDataset('output/{}/raw_audio.nc'.format(corpus))
56+
dataset = LabelledDataset('datasets/{}/files.txt'.format(corpus))
5957
dataset.pad_arrays()
6058

6159
class_weight = (dataset.n_instances
@@ -80,7 +78,7 @@ def main():
8078
df = within_corpus_cross_validation(clf, dataset,
8179
splitter=LeaveOneGroupOut())
8280
print_results(df)
83-
output_dir = Path(RESULTS_DIR) / corpus
81+
output_dir = Path('results') / 'latif2019' / corpus
8482
output_dir.mkdir(parents=True, exist_ok=True)
8583
df.to_csv(output_dir / 'raw_audio.csv')
8684

0 commit comments

Comments
 (0)