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)

After Change

    viz.plot_experiment(experiment.spec, experiment_df, METRICS_COLS)
    logger.debug(f"Saved experiment data to {prepath}")
    // zip files
    predir, _, _, _, _, _ = util.prepath_split(prepath)
    shutil.make_archive(predir, "zip", predir)
    logger.info(f"All experiment data zipped to {predir}.zip")
    return experiment_df

Instances