tensorflow implementation for TATE
tensorflow-gpu==1.14.0
torch == 1.2.0
transformers==2.10.0
numpy==1.15.4
sklearn
python 3.6.13
-
download processed features from here , code: 2qiu
-
or process file by OpenSimle2.0, Bert, and Librosa. Refer to utils folder.
-
generate train.pkl and test.pkl
python Dataset.py- train (can modify settings in Settings.py)
python train.py -
As for the IEMOCAP dataset, replacing the Datset.py, network.py and train.py in the iemocap folder.
-
test
python train.py --train=False