readd platt scaling

EleutherAI · Aug 23, 2023 · 89f8987 · 89f8987
1 parent 777912e
commit 89f8987
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/elk/training/train.py b/elk/training/train.py
@@ -84,6 +84,7 @@ def apply_to_layer(
  train_loss = reporter.fit(first_train_h)
  labels = to_one_hot(train_gt, k)
  labels = repeat(labels, "n k -> n v k", v=v)
+ reporter.platt_scale(labels, first_train_h)
 
  elif isinstance(self.net, EigenFitterConfig):
  fitter = EigenFitter(