Aktualisieren models.py

models.py

@@ -60,6 +60,8 @@ class SummarisationModel(nn.Module):
             return probs.topk(k).indices, probs # handle doc weniger als 3 sents? 
         return torch.arange(len(probs)), probs
 
+    def test(self, dataset):
+
 
 
 class ActorOnlySummarisationModel(SummarisationModel):