paper preperations and notebooks, optuna callbacks

2021-04-02 08:45:11 +02:00
parent 7c88602776
commit cec3a07d60
21 changed files with 3818 additions and 1059 deletions
--- a/models/testing.py
+++ b/models/testing.py
@ -109,9 +109,3 @@ class Tester(CombinedModelMixins,
        tensor = self.to_cls_token(tensor[:, 0])
        tensor = self.mlp_head(tensor)
        return Namespace(main_out=tensor, attn_weights=None)
-
-    def additional_scores(self, outputs):
-        if self.params.n_classes > 2:
-            return MultiClassScores(self)(outputs)
-        else:
-            return BinaryScores(self)(outputs)