refactor and combined_botorch_baybe...

AC-BO-Hackathon · Mar 30, 2024 · 57b8ea9 · 57b8ea9
1 parent f78df9a
commit 57b8ea9
Show file tree

Hide file tree

Showing 11 changed files with 391 additions and 4 deletions.
diff --git a/BoTorch_heatmapTrue.png b/BoTorch_heatmapTrue.png
diff --git a/BoTorch_heatmap_delta.png b/BoTorch_heatmap_delta.png
diff --git a/combined_botorch_baybe_final.ipynb b/combined_botorch_baybe_final.ipynb
diff --git a/results.zip b/results.zip
diff --git a/run_experiment.py b/run_experiment.py
@@ -149,8 +149,8 @@ def run_experiment(n_init, noise_level, budget, seed, noise_bool):
     mll, model = initialize_model(train_x, train_Y, noise_bool)
     fit_gpytorch_model(mll)
 
-    os.makedirs('results', exist_ok=True)
-    fname = f"results/{problem.__class__.__name__[:5]}_n_init_{n_init}_noiselvl_{noise_level}_budget_{budget}_seed_{seed}_noise_{noise_bool}.pt"
+    os.makedirs('results_botorch', exist_ok=True)
+    fname = f"results_botorch/{problem.__class__.__name__[:5]}_n_init_{n_init}_noiselvl_{noise_level}_budget_{budget}_seed_{seed}_noise_{noise_bool}.pt"
     torch.save((train_X, train_Y, train_Y_real, model), fname)
 
     return train_X, train_Y, train_Y_real, model

diff --git a/run_experiment_baybe.py b/run_experiment_baybe.py
@@ -125,8 +125,8 @@ def run_experiment(n_init, noise_level, budget, seed, noise_bool):
     train_Y = torch.from_numpy(y_train)
     train_Y_real = torch.from_numpy(y_real_complete)
 
-    os.makedirs('results', exist_ok=True)
-    fname = f"results/{problem.__class__.__name__[:5]}_n_init_{n_init}_noiselvl_{noise_level}_budget_{budget}_seed_{seed}_noise_{noise_bool}.pt"
+    os.makedirs('results_baybe', exist_ok=True)
+    fname = f"results_baybe/{problem.__class__.__name__[:5]}_n_init_{n_init}_noiselvl_{noise_level}_budget_{budget}_seed_{seed}_noise_{noise_bool}.pt"
     torch.save((train_X, train_Y, train_Y_real, None), fname)
 
     return train_X, train_Y, train_Y_real, None

diff --git a/run_grid_botorch.py b/run_grid_botorch.py
@@ -0,0 +1,16 @@
+import torch
+import pandas as pd
+from run_grid_experiments import run_grid_experiments
+from run_experiment import run_experiment
+
+
+seeds = list(range(5))
+n_inits = [2, 4, 8, 10]
+noise_levels = [1, 5, 10, 20]
+noise_bools = [True, False]
+budget = 30
+
+
+run_grid_experiments(seeds, n_inits, noise_levels, noise_bools, budget)
+
+
diff --git a/run_grid_experiments.py b/run_grid_experiments.py
@@ -1,10 +1,12 @@
 import ray
 import argparse
 from time import time, sleep
+import sys
 from run_experiment import run_experiment
 from datetime import datetime
 import gc
 
+
 MAX_NUM_PENDING_TASKS = 12
 
 

diff --git a/botorch_version_grid_serach_try.ipynb → ...oks/botorch_version_grid_serach_try.ipynb b/botorch_version_grid_serach_try.ipynb → ...oks/botorch_version_grid_serach_try.ipynb
diff --git a/gpax_schwefel.ipynb → trial_notebooks/gpax_schwefel.ipynb b/gpax_schwefel.ipynb → trial_notebooks/gpax_schwefel.ipynb
diff --git a/gpax_schwefel_beta.ipynb → trial_notebooks/gpax_schwefel_beta.ipynb b/gpax_schwefel_beta.ipynb → trial_notebooks/gpax_schwefel_beta.ipynb