def test_extreme_values(self):
S, y = get_extreme_binary_example()
self.cal = TemperatureScaling(logit_constant=0.0)
self.cal.fit(S, y)
predictions = self.cal.predict_proba(S).argmax(axis=1)
acc = accuracy_score(y, predictions)
self.assertGreater(acc, 0.99,
"accuracy must be superior to 99 percent")
def setUp(self):
self.cal = TemperatureScaling(logit_constant=0.0)