| 39 | fit(args) |
| 40 | |
| 41 | def fit_sh(): |
| 42 | for poly in POLYS: |
| 43 | for calc in CALC: |
| 44 | calc_name = calc.replace("-","_") # closed-form to closed_form |
| 45 | |
| 46 | pe = 'sphericalharmonics' |
| 47 | args = Namespace(dataset='checkerboard', |
| 48 | pe=pe, |
| 49 | nn=NN, |
| 50 | save_model=False, |
| 51 | log_wandb=False, |
| 52 | hparams='hparams.yaml', |
| 53 | results_dir='results/train/exp_computation', |
| 54 | expname=f'{calc_name}-{PE}-{poly}poly', |
| 55 | seed=0, |
| 56 | harmonics_calculation=calc, |
| 57 | resume_ckpt_from_results_dir=False, |
| 58 | matplotlib=True, |
| 59 | matplotlib_show=False, |
| 60 | checkerboard_scale=1, |
| 61 | legendre_polys=poly, |
| 62 | min_radius=None) |
| 63 | |
| 64 | fit(args) |
| 65 | return args |
| 66 | |
| 67 | |
| 68 | def extract_sh_df(results_dir, dataset): |