Skip to content

Commit 099b666

Browse files
committed
added explicit classifier column to scoring dataframes and updated gather results accordingly
1 parent 74fe9d0 commit 099b666

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

scripts/shapeembed/evaluation.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,7 +105,8 @@ def score_dataframe( df, name
105105
# Put the results into a DataFrame
106106
df = pandas.DataFrame(cv_results)
107107
df = df.drop(["fit_time", "score_time"], axis=1)
108-
df.insert(loc=0, column='trial', value=f'{name}_{pipename}')
108+
df.insert(loc=0, column='classifier', value=pipename)
109+
df.insert(loc=0, column='trial', value=name)
109110
tag_columns.reverse()
110111
for tag_col_name, tag_col_value in tag_columns:
111112
df.insert(loc=0, column=tag_col_name, value=tag_col_value)

scripts/shapeembed/gather_run_results.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -142,7 +142,7 @@ def main_process(clargs, logger=logging.getLogger(__name__)):
142142
# function for finding total
143143
def keep_first_fname(series):
144144
return functools.reduce(lambda x, y: y if str(x) == 'nofile' else x, series)
145-
idx_cols = ['trial', 'dataset', 'model', 'compression_factor', 'latent_dim', 'batch_size']
145+
idx_cols = ['trial', 'classifier', 'dataset', 'model', 'compression_factor', 'latent_dim', 'batch_size']
146146
df.set_index(idx_cols, inplace=True)
147147
df.sort_index(inplace=True)
148148
#df = df.groupby(level=['trial', 'dataset', 'model', 'compression_factor', 'latent_dim', 'batch_size']).agg({

0 commit comments

Comments
 (0)