Code for future talk move prediction from the ACL 2021 Findings paper "What Would a Teacher Do? Predicting Future Talk Moves"
Model settings can be configured in experiments.conf
, including specifying if pretrained layers need to be used.
To use the GPU, change device from 'cpu' to 'cuda' in train.py
, models.py
and nn_blocks.py
Train models using python train.py --expr name_of_experiment
, e.g. python train.py --expr retrain
. Specify the location of where to save the model in log_dir
of the experiment configuration.
Evaluate models using python evaluation.py name_of_experiment
.