9da4e51da0044cfbdcf3a667187d86f0c35ece8e,slm_lab/experiment/analysis.py,,analyze_experiment,#Any#,301

Before Change

    @returns {DataFrame} experiment_df Of var_specs, metrics for all trials.
    """
    experiment_df = pd.DataFrame(experiment.trial_data_dict).transpose()
    cols = METRICS_COLS
    config_cols = sorted(ps.difference(experiment_df.columns.tolist(), cols))
    sorted_cols = config_cols + cols
    experiment_df = experiment_df.reindex(sorted_cols, axis=1)
    experiment_df.sort_values(by=["strength"], ascending=False, inplace=True)
    logger.info(f"Experiment data:\n{experiment_df}")
    experiment_fig = plot_experiment(experiment.spec, experiment_df)
    save_experiment_data(experiment.spec, experiment_df, experiment_fig)
    return experiment_df

After Change

def analyze_experiment(experiment):
    """Analyze experiment and save data"""
    prepath = experiment.spec["meta"]["prepath"]
    // calculate experiment df
    experiment_df = calc_experiment_df(experiment.trial_data_dict, prepath)
    // plot graph

Instances