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