trial function, plan to integrate inspyredtk

jchen6727 · May 21, 2024 · 52e3032 · 52e3032
1 parent a729916
commit 52e3032
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 4 deletions.
diff --git a/batchtk/inspyredtk/__init__.py b/batchtk/inspyredtk/__init__.py
@@ -0,0 +1,3 @@
+
+
+
diff --git a/batchtk/inspyredtk/search.py b/batchtk/inspyredtk/search.py
@@ -0,0 +1,10 @@
+from inspyred import ec
+import random
+
+ALGO = {
+    'DEA': ec.DEA,
+    'EDA': ec.EDA,
+    'ES': ec.ES,
+    'GA': ec.GA,
+    'SA': ec.SA,
+}
diff --git a/batchtk/runtk/trial.py b/batchtk/runtk/trial.py
@@ -1,12 +1,21 @@
-def trials():
-    pass
+import types
+import pandas
+def trials(configs, label, gen, dispatcher_constructor, project_path, output_path, submit):
+    results = []
+    for tid, config in enumerate(configs):
+        results.append(trial(config, label, tid, dispatcher_constructor, project_path, output_path, submit))
+    return results
 
 
 def trial(config, label, tid, dispatcher_constructor, project_path, output_path, submit):
     run_label = '{}_{}'.format(label, tid)
+    trial.run_label = run_label
+    trial.output_path = output_path
+    for k, v in config.items(): #call any function pointers
+        if isinstance(v, types.FunctionType):
+            config[k] = v()
     dispatcher = dispatcher_constructor(project_path=project_path, output_path=output_path, submit=submit,
                                         gid=run_label)
-
     dispatcher.update_env(dictionary=config)
     try:
         dispatcher.run()
@@ -17,4 +26,10 @@ def trial(config, label, tid, dispatcher_constructor, project_path, output_path,
         dispatcher.clean()
         raise (e)
     data = pandas.read_json(data, typ='series', dtype=float)
-    return data
+    return data
+
+
+LABEL_POINTER = lambda:trial.run_label
+
+
+PATH_POINTER = lambda:trial.output_path