As shown in Figure 1, we present the calibration of a BERT-MLP model for a text classification task on the 20NG dataset.
we train a TextCNN (Kim, 2014) and a BERT-MLP using 20NG 15 (the first 15 categories of 20NG) and then evaluate them on both in-distribution and OOD data.