examples/research_projects/bert-loses-patience/pabee/modeling_pabee_bert.py/BertModelWithPabee/set_regression_threshold
class BertModelWithPabee:    def set_regression_threshold(self, threshold):
        self.regression_threshold = threshold