Skip to content

Commit

Permalink
Small typo fix to examples in cookbook which had resulted in an impor…
Browse files Browse the repository at this point in the history
…t error
  • Loading branch information
mkretsch327 authored and rlouf committed Apr 1, 2024
1 parent aed9d21 commit 1d20896
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/cookbook/classification.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,7 @@ def customer_support(request):
Outlines provides a shortcut to do multi-label classification, using the `outlines.generate.choice` function to initialize a generator. Outlines uses multinomial sampling by default, here we will use the greedy sampler to get the label with the highest probability:

```python
from outlines.generate.samplers import greedy
from outlines.samplers import greedy

generator = outlines.generate.choice(model, ["URGENT", "STANDARD"], sampler=greedy)
```
Expand Down

0 comments on commit 1d20896

Please sign in to comment.