1111import tensorflow as tf
1212from 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
1515from emotion_recognition .tensorflow .classification import TFClassifier
1616from emotion_recognition .tensorflow .models import latif2019_model
1717from emotion_recognition .tensorflow .utils import create_tf_dataset_ragged
1818from sklearn .model_selection import LeaveOneGroupOut
1919from tensorflow .keras .callbacks import EarlyStopping , ReduceLROnPlateau
2020from tensorflow .keras .optimizers import RMSprop
2121
22- RESULTS_DIR = 'results/latif2019'
23-
2422
2523def 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