1
0
Fork 0
mirror of https://github.com/Findus23/rebound-collisions.git synced 2024-09-19 15:53:48 +02:00

add script printing average CPU hours per simulations

This commit is contained in:
Lukas Winkler 2022-02-27 15:15:25 +01:00
parent 8243336ada
commit 28aa9555bb
Signed by: lukas
GPG key ID: 54DE4D798D244853

26
runtime_stats.py Normal file
View file

@ -0,0 +1,26 @@
from pathlib import Path
import numpy as np
from scipy.constants import hour
from extradata import ExtraData
from utils import filename_from_argv, mode_from_fn
files = Path("data/").glob("final*.bin")
times = []
for file in files:
fn = filename_from_argv(str(file))
mode = mode_from_fn(fn)
if mode == "lz":
continue
if mode != "lz_correct":
continue
print(fn)
ed = ExtraData.load(fn)
if str(fn) == "data/final_lz_correct_15":
print(ed.meta.cputime / hour)
times.append(ed.meta.cputime)
times = np.asarray(times)
print(times.mean() / hour)
print(times.std() / hour)