This repository contains the source code and data for the paper: DialBERT: A Hierarchical Pre-Trained Model for Conversation Disentanglement https://arxiv.org/abs/2004.03760
Python 3.6 Tensorflow 1.14.0
This is Adaptation model
https://drive.google.com/drive/folders/1n6ymYqM1xv08mWW0I-kHIBylK4VR6W-F?usp=sharing
This is BERT-base Pre-trained checkpoints
https://drive.google.com/drive/folders/13Q0OtsRKSAEilt4U-ycQBa4CWFsGHbpv?usp=sharingcd ..
cd Disentangle/scripts/bash_file/
bash train_exloss.sh
cd Disentangle/scripts/ckpt/your_ckpt
base valid.sh
@article{DBLP:journals/corr/abs-2004-03760,
author = {Tianda Li and
Jia{-}Chen Gu and
Xiaodan Zhu and
Quan Liu and
Zhen{-}Hua Ling and
Zhiming Su and
Si Wei},
title = {DialBERT: {A} Hierarchical Pre-Trained Model for Conversation Disentanglement},
journal = {CoRR},
volume = {abs/2004.03760},
year = {2020},
url = {https://arxiv.org/abs/2004.03760},
archivePrefix = {arXiv},
eprint = {2004.03760},
timestamp = {Tue, 14 Apr 2020 16:40:34 +0200},
biburl = {https://dblp.org/rec/journals/corr/abs-2004-03760.bib},
bibsource = {dblp computer science bibliography, https://dblp.org}
}