From 1907c040adb6321119455575ea8435c4226dc251 Mon Sep 17 00:00:00 2001 From: Alejandro Moreo Date: Tue, 3 Mar 2026 10:06:08 +0100 Subject: [PATCH] added total to lequa protocols for bag iteration --- quapy/data/_lequa.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/quapy/data/_lequa.py b/quapy/data/_lequa.py index e162f4c..7e4cf52 100644 --- a/quapy/data/_lequa.py +++ b/quapy/data/_lequa.py @@ -99,6 +99,9 @@ class SamplesFromDir(AbstractProtocol): sample, _ = self.load_fn(os.path.join(self.path_dir, f'{id}.txt')) yield sample, prevalence + def total(self): + return len(self.true_prevs) + class LabelledCollectionsFromDir(AbstractProtocol): @@ -113,6 +116,10 @@ class LabelledCollectionsFromDir(AbstractProtocol): lc = LabelledCollection.load(path=collection_path, loader_func=self.load_fn) yield lc + def total(self): + return len(self.true_prevs) + + class ResultSubmission: