Rename example usage files

aakamishra · Oct 23, 2022 · 6d87ca0 · 6d87ca0
1 parent e5eb842
commit 6d87ca0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/test_experiments.py → experiments_example.py b/test_experiments.py → experiments_example.py
@@ -1,16 +1,14 @@
-import phoss
 from phoss.experiment_analysis import ExperimentAnalysis
 from phoss.experiment_group import ExperimentGroup
 
-seeds = [61, 121, 124, 161, 165]
-# methods = [ 'Random', 'ASHA', 'Hyperband', 'PTB']
-methods = ['Random', 'ASHA', 'Hyperband']
+seeds = [161, 165]
+methods = ['Random', 'ASHA', 'Hyperband', 'PBT']
 results = []
 exp_groups = []
 save_dir = 'checkpoints'
 num_workers = 16
 num_samples = 100
-max_num_epochs = 100
+max_num_epochs = 25
 
 for method in methods:
     egroup = ExperimentGroup(
@@ -21,7 +19,7 @@
         num_actors=num_workers,
         num_samples=num_samples,
         save_dir='checkpoints',
-        verbose=1
+        verbose=0
     )
     result = egroup.run()
     exp_groups.append(egroup)

diff --git a/test_landscaper.py → landscaper_example.py b/test_landscaper.py → landscaper_example.py
@@ -20,6 +20,8 @@
 plt.plot(time_range, sim_loss, alpha=0.1, color='blue')
 plt.plot(time_range, landscaper.true_loss, alpha=0.1, color='red')
 plt.show()
+plt.close()
 
 sns.heatmap(sim_loss)
 plt.show()
+plt.close()