noahjadallah/cause-effect-detection

1---2widget:3- text: "If a user signs up, he will receive a confirmation email."4---5 6# Cause-Effect Detection for Software Requirements Based on Token Classification with BERT7 8This model uses BERT to detect cause and effect from a single sentence. The focus of this model is the domain of software requirements engineering, however, it can also be used for other domains.9 10The model outputs one of the following 5 labels for each token:11 12Other13 14B-Cause15 16I-Cause17 18B-Effect19 20I-Effect21 22The source code can be found here: https://colab.research.google.com/drive/14V9Ooy3aNPsRfTK88krwsereia8cfSPc?usp=sharing