Function main

experiments/exp_computation.py:163–188

()

    plt.show()

def main():
    results_dir = 'results/train/exp_computation'
    dataset = 'checkerboard'
    # args = fit_sh()
    #fit_comparison()

    df = extract_comp_df(results_dir, dataset)
    plot_comp(df, ydim="test_duration")
    plot_comp(df, ydim="accuracy")

    for pe in df.pe.unique():
        #for poly in df.poly.unique(): (df.poly == poly) &
        df.loc[(df.pe == pe)].sort_values(by="poly").to_csv(os.path.join(results_dir, f"{pe}.csv"))


    df = extract_sh_df(results_dir, dataset)
    plot_sh(df, ydim="test_duration")
    plot_sh(df, ydim="accuracy")

    for comp in df.harmonics_calculation.unique():
        #for poly in df.poly.unique(): (df.poly == poly) &
        df.loc[(df.harmonics_calculation == comp)].sort_values(by="poly").to_csv(os.path.join(results_dir, f"sphericalharmonics-{comp}.csv"))

    #for nn in df.nn.unique():
    #    for poly in df.poly.unique():
    #        df.loc[(df.poly == poly) & (df.nn == nn)].sort_values(by="mean_dist").to_csv(os.path.join(resultsdir, f"sphericalharmonics-{nn}-{poly}.csv"))

if __name__ == '__main__':
    main()