Skip to content

Commit

Permalink
feat(readme): update custom scoring function documentation
Browse files Browse the repository at this point in the history
  • Loading branch information
Juanpablor5 committed Oct 4, 2024
1 parent 3810b00 commit 6be518f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -86,13 +86,13 @@ naml = naiveautoml.NaiveAutoML(scoring="accuracy", passive_scorings=["neg_log_lo
You can also pass a custom scoring function through a dictionary:

```python
scorer = make_scorer(**{
scorer = {
"name": "accuracy",
"score_func": lambda y, y_pred: np.count_nonzero(y == y_pred).mean(),
"greater_is_better": True,
"needs_proba": False,
"needs_threshold": False
})
}
naml = naiveautoml.NaiveAutoML(scoring=scorer)
```

Expand Down

0 comments on commit 6be518f

Please sign in to comment.