Some small code improvements

ml6team · Jan 10, 2024 · dce26fd · dce26fd
1 parent f9067c8
commit dce26fd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/parameter_search.ipynb b/src/parameter_search.ipynb
@@ -306,14 +306,14 @@
    "outputs": [],
    "source": [
     "searchable_index_params = {\n",
-    "    'chunk_size' : [384],\n",
-    "    'chunk_overlap' : [128],\n",
+    "    'chunk_size' : [192, 256, 320],\n",
+    "    'chunk_overlap' : [64, 128, 192],\n",
     "}\n",
     "searchable_shared_params = {\n",
-    "    'embed_model' : [(\"huggingface\", \"BAAI/bge-base-en-v1.5\")]\n",
+    "    'embed_model' : [(\"huggingface\",\"all-MiniLM-L6-v2\"), (\"huggingface\", \"BAAI/bge-base-en-v1.5\")]\n",
     "}\n",
     "searchable_eval_params = {\n",
-    "    'retrieval_top_k' : [2]\n",
+    "    'retrieval_top_k' : [2, 4, 8]\n",
     "}\n",
     "\n",
     "search_method = 'progressive_search' # 'grid_search', 'progressive_search'\n",

diff --git a/src/utils.py b/src/utils.py
@@ -203,7 +203,7 @@ def create_configs(self, runcount):
 
             # if there are no variations to try, just schedule one run
             if len(variations_to_try) == 0:
-                variations_to_try = [{next(iter(pipeline_config.items()))[0] : next(iter(pipeline_config.items()))[1]}]
+                variations_to_try = [{list(pipeline_config.keys())[0]: list(pipeline_config.values())[0]}]
 
             # when all variations have been tried, stop searching
             if runcount > len(variations_to_try) - 1: