Skip to content

Commit 71d2507

Browse files
committed
use score-based inconsistency resolution
1 parent c220b8c commit 71d2507

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

chebifier/ensemble/base_ensemble.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from chebifier.check_env import check_package_installed
1111
from chebifier.hugging_face import download_model_files
12-
from chebifier.inconsistency_resolution import PredictionSmoother
12+
from chebifier.inconsistency_resolution import ScoreBasedPredictionSmoother
1313
from chebifier.prediction_models.base_predictor import BasePredictor
1414
from chebifier.utils import (
1515
get_default_configs,
@@ -75,7 +75,7 @@ def __init__(
7575
self.models.append(model_instance)
7676

7777
if resolve_inconsistencies:
78-
self.smoother = PredictionSmoother(
78+
self.smoother = ScoreBasedPredictionSmoother(
7979
self.chebi_graph,
8080
label_names=None,
8181
disjoint_files=self.disjoint_files,

0 commit comments

Comments
 (0)