Skip to content

Commit 5e6ca07

Browse files
committed
refactor: rename binary weighted knn explainer game
1 parent 2030858 commit 5e6ca07

File tree

1 file changed

+2
-2
lines changed
  • src/shapiq/explainer/nn/games

1 file changed

+2
-2
lines changed

src/shapiq/explainer/nn/games/wknn.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def __init__(
4545
raise ValueError(msg)
4646

4747
self.binary_games = {
48-
other_class_index: BinaryWeightedKNNExplainerXAI(
48+
other_class_index: BinaryWeightedKNNExplainerGame(
4949
model, x, class_index, other_class_index, n_bits
5050
)
5151
for other_class_index in range(self.n_classes)
@@ -61,7 +61,7 @@ def value_function(self, coalitions: npt.NDArray[np.bool]) -> GameValues:
6161
return sum(binary_game_values) / (self.n_classes - 1)
6262

6363

64-
class BinaryWeightedKNNExplainerXAI(KNNExplainerGameBase):
64+
class BinaryWeightedKNNExplainerGame(KNNExplainerGameBase):
6565
"""Benchmark game for the WeightedKNNExplainer in the binary case."""
6666

6767
def __init__(

0 commit comments

Comments
 (0)