Update train_optillm_classifier.py

codelion · codelion · commit 12c7f636a6f6 · 2024-10-24T14:37:32.000+08:00
diff --git a/scripts/train_optillm_classifier.py b/scripts/train_optillm_classifier.py
@@ -297,7 +297,7 @@ def main(args):
         "Find all possible combinations of coins that sum up to $1"
     ]
 
-    effort_levels = [0.2, 0.5, 0.8, 1.0]
+    effort_levels = [0.0, 0.2, 0.5, 0.8, 1.0]
 
     print("\nInference Examples:")
     for prompt in test_prompts:

Original file line number	Diff line number	Diff line change
`@@ -297,7 +297,7 @@ def main(args):`
`297`	`297`	`"Find all possible combinations of coins that sum up to $1"`
`298`	`298`	`]`
`299`	`299`
`300`		`- effort_levels = [0.2, 0.5, 0.8, 1.0]`
	`300`	`+ effort_levels = [0.0, 0.2, 0.5, 0.8, 1.0]`
`301`	`301`
`302`	`302`	`print("\nInference Examples:")`
`303`	`303`	`for prompt in test_prompts:`