TATE

tensorflow implementation for TATE

Requirements

tensorflow-gpu==1.14.0
torch == 1.2.0
transformers==2.10.0
numpy==1.15.4
sklearn
python 3.6.13

Usage

• download processed features from Google Drive or Baidu Disk

• 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