Merge branch 'master' of github.com:nikhilpim/medley-solver into TACA…

…S2021
uclid-org · Oct 13, 2020 · 2b99bac · 2b99bac
2 parents 62836f9 + b1b89ad
commit 2b99bac
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/bin/medley b/bin/medley
@@ -165,6 +165,13 @@ def main():
         default="both"
     )
 
+    global_parser.add_argument(
+        "--gamma",
+        help="gamma parameter for exp3",
+        type=float,
+        default=0.07
+    )
+
     args = global_parser.parse_args()
 
     if not args.output.endswith(".csv"):
@@ -209,7 +216,7 @@ def main():
         classifier = Preset(args.preset)
         timeout_manager = Constant(args.timeout)
     elif args.classifier == "exp3":
-        classifier = Exp3(0.07)
+        classifier = Exp3(args.gamma)
     elif args.classifier == "knearest":
         classifier = KNearest(args.k, args.epsilon, args.epsilon_decay, args.time_k)
     else: