Upd

krasheninnikov · Feb 22, 2024 · f681e58 · f681e58
1 parent 2564d5b
commit f681e58
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/configs/current_experiment.yaml b/configs/current_experiment.yaml
@@ -61,7 +61,7 @@ numeric_experiment_arguments:
   fn_input_len: 4
   nfunc: 20
   n_fns_to_lock: 5
-  n_fns_to_unlock: 1
+  n_fns_to_unlock: 4
 
 
 # overrides specified parameters

diff --git a/utils/aggregation_utils.py b/utils/aggregation_utils.py
@@ -13,7 +13,7 @@
 rc('text', usetex=True)
 plt.rcParams['text.usetex'] = True
 
-rc('text.latex', preamble=r'\usepackage{color, amsfonts, amsmath, amsthm}')
+# rc('text.latex', preamble=r'\usepackage{color, amsfonts, amsmath, amsthm}')
 
 
 def aggregate_mean_std_count(df):
@@ -134,7 +134,7 @@ def prettify_label(label):
 
 def make_experiment_plot(exp_name, stage_paths, thruncate_stages_after_epoch=None, eval_each_epochs_per_stage=None,
                          tags=['eval/d1consis_EM', 'eval/d2consis_EM'], os_list=None, ylabel='Value', title='',
-                         figsize=(5.7,4), legend_loc='best', colors=None):
+                         figsize=(5.7,4), legend_loc='best', colors=None, no_plot=False):
     """
     exp_name - name of the experiment (top level folder name)
     stage_paths - list of strings that are the starts to paths to stages, 
@@ -227,6 +227,9 @@ def make_experiment_plot(exp_name, stage_paths, thruncate_stages_after_epoch=Non
 
     df = pd.concat(dfs_all_stages, axis=0)
 
+    if no_plot:
+        return df
+
     # add a column with log of value
     # df['log_value'] = np.log(df['value'])
     df['tag'] = df['tag'].apply(lambda x: x.replace('eval/', '').replace('train_', '').replace('_EM', '').replace('_loss', ''))