From b1b89ada736d62f7291b375cc86d75e1d4bf365b Mon Sep 17 00:00:00 2001
From: Nikhil Pimpalkhare <nikhilpimpalkhare@Nikhils-MacBook-Pro.local>
Date: Tue, 13 Oct 2020 14:43:46 -0700
Subject: [PATCH] added gamma option

---
 bin/medley | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/bin/medley b/bin/medley
index 27b39d3..5fe5554 100755
--- a/bin/medley
+++ b/bin/medley
@@ -134,6 +134,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"):
@@ -174,7 +181,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)
     else: