Skip to content

Commit 0011248

Browse files
committed
Fix example in readme
1 parent b04066a commit 0011248

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ Data Shapley values:
9999
```python
100100
import numpy as np
101101
from pydvl.utils import Dataset, Utility
102-
from pydvl.value.shapley import compute_shapley_values
102+
from pydvl.value import *
103103
from sklearn.linear_model import LinearRegression
104104
from sklearn.model_selection import train_test_split
105105

@@ -111,7 +111,7 @@ dataset = Dataset(X_train, y_train, X_test, y_test)
111111
model = LinearRegression()
112112
utility = Utility(model, dataset)
113113
values = compute_shapley_values(
114-
u=utility, n_iterations=100, mode="truncated_montecarlo"
114+
u=utility, mode="truncated_montecarlo", done=MaxUpdates(100)
115115
)
116116
```
117117

0 commit comments

Comments
 (0)