mirror of
https://github.com/p-e-w/heretic.git
synced 2026-09-25 13:31:33 -07:00
feat: add benchmark scorer (#444)
* fix: improve print output of scorers * feat: add benchmark scorer
This commit is contained in:
@@ -9,7 +9,6 @@ print_debug_information = true
|
||||
|
||||
batch_size = 2
|
||||
max_response_length = 10
|
||||
kl_divergence_target = 0
|
||||
n_trials = 2
|
||||
n_startup_trials = 1
|
||||
|
||||
@@ -21,11 +20,6 @@ save_directory = "model"
|
||||
|
||||
row_normalization = "pre"
|
||||
|
||||
scorers = [
|
||||
{ plugin = "heretic.scorers.keyword_rate.KeywordRate", optimization = "minimize" },
|
||||
{ plugin = "heretic.scorers.kl_divergence.KLDivergence", optimization = "minimize" },
|
||||
]
|
||||
|
||||
[good_prompts]
|
||||
dataset = "mlabonne/harmless_alpaca"
|
||||
commit = "02c6a92cfcf11bb0c387334f8146d149d65b587f"
|
||||
|
||||
Reference in New Issue
Block a user