From 11ed5953e45375b72d76d8b819f34f3b825fd0ab Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Sun, 11 Jun 2023 22:37:32 +0800 Subject: [PATCH] [MNT] add two arguments to abduce_pseudo_label in SimpleBridge --- abl/bridge/simple_bridge.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/abl/bridge/simple_bridge.py b/abl/bridge/simple_bridge.py index 806733e..8b71d9a 100644 --- a/abl/bridge/simple_bridge.py +++ b/abl/bridge/simple_bridge.py @@ -24,15 +24,17 @@ class SimpleBridge(BaseBridge): pred_res = self.model.predict(X) pred_label, pred_prob = pred_res["label"], pred_res["prob"] return pred_label, pred_prob - + def abduce_pseudo_label( self, pred_label: List[List[Any]], pred_prob: ndarray, pseudo_label: List[List[Any]], Y: List[List[Any]], + max_revision: int = -1, + require_more_revision: int = 0, ) -> List[List[Any]]: - return self.abducer.batch_abduce(pred_label, pred_prob, pseudo_label, Y) + return self.abducer.batch_abduce(pred_label, pred_prob, pseudo_label, Y, max_revision, require_more_revision) def label_to_pseudo_label( self, label: List[List[Any]], mapping: Dict = None