initial commit for pdtb pretraining task
@@ -27,7 +27,7 @@ from tasks import SingleClassificationTask, PairClassificationTask, \
MultiNLISlateTask, MultiNLIGovernmentTask, MultiNLITravelTask, \
MultiNLITelephoneTask, QQPTask, RTETask, \
QNLITask, SNLITask, SSTTask, STSBTask, WNLITask, \
- LanguageModelingTask, WikiTextLMTask
+ LanguageModelingTask, WikiTextLMTask, PDTBTask
NAME2INFO = {'sst': (SSTTask, 'SST-2/'),
'cola': (CoLATask, 'CoLA/'),
@@ -45,6 +45,7 @@ NAME2INFO = {'sst': (SSTTask, 'SST-2/'),
'snli': (SNLITask, 'SNLI/'),
'wnli': (WNLITask, 'WNLI/'),
'wiki': (WikiTextLMTask, 'WikiText/')
+ 'pdtb': (PDTBTask, 'PDTB/')
}
