Skip to content

Commit 2038788

Browse files
authored
fix doc (#232)
1 parent 38fda07 commit 2038788

File tree

2 files changed

+12
-0
lines changed

2 files changed

+12
-0
lines changed

dattri/algorithm/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
"""dattri.algorithm for some data attribution methods."""
22

33
from .block_projected_if.block_projected_if import BlockProjectedIFAttributor
4+
from .data_shapley import KNNShapleyAttributor
45
from .factgrass import FactGraSSAttributor
56
from .influence_function import (
67
IFAttributorArnoldi,
@@ -24,6 +25,7 @@
2425
"IFAttributorEKFAC",
2526
"IFAttributorExplicit",
2627
"IFAttributorLiSSA",
28+
"KNNShapleyAttributor",
2729
"LoGraAttributor",
2830
"RPSAttributor",
2931
"TRAKAttributor",

docs/source/api/algorithm.rst

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,3 +31,13 @@ Attributors
3131
:special-members: __init__
3232
:undoc-members:
3333
:show-inheritance:
34+
.. autoclass:: dattri.algorithm.block_projected_if.BlockProjectedIFAttributor
35+
:members:
36+
:special-members: __init__
37+
:undoc-members:
38+
:show-inheritance:
39+
.. autoclass:: dattri.algorithm.factgrass.FactGraSSAttributor
40+
:members:
41+
:special-members: __init__
42+
:undoc-members:
43+
:show-inheritance:

0 commit comments

Comments
 (0)