We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 79fdfdd commit fb3aa86Copy full SHA for fb3aa86
1 file changed
delphi/scorers/classifier/classifier.py
@@ -22,6 +22,7 @@ def __init__(
22
verbose: bool,
23
n_examples_shown: int,
24
seed: int = 42,
25
+ log_prob: bool = False,
26
**generation_kwargs,
27
):
28
"""
@@ -41,7 +42,7 @@ def __init__(
41
42
self.verbose = verbose
43
self.n_examples_shown = n_examples_shown
44
self.generation_kwargs = generation_kwargs
- self.log_prob = False
45
+ self.log_prob = log_prob
46
self.rng = random.Random(seed)
47
48
async def __call__(
0 commit comments