mirror/halo_comparison
1
0
Fork 0
mirror of https://github.com/Findus23/halo_comparison.git synced 2024-09-19 16:03:50 +02:00
Code Releases Activity

speed up similarity calculation

Browse source
This commit is contained in:
Lukas Winkler 2022-07-20 15:21:35 +02:00
parent 22c537aeec
commit 15b43bb7c1
Signed by: lukas
GPG key ID: 54DE4D798D244853
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
compare_halos.py
View file

@ -235,9 +235,12 @@ def compare_halo_resolutions(
halo_size = len(particle_ids_in_comp_halo) halo_size = len(particle_ids_in_comp_halo)
# df = particles_in_comp_halo.join(halo_particles, how="inner", rsuffix="ref") # df = particles_in_comp_halo.join(halo_particles, how="inner", rsuffix="ref")
shared_particles = particle_ids_in_comp_halo.intersection(halo_particle_ids) shared_particles = particle_ids_in_comp_halo.intersection(halo_particle_ids)
union_particles = particle_ids_in_comp_halo.union(halo_particle_ids) # union_particles = particle_ids_in_comp_halo.union(halo_particle_ids)
similarity = len(shared_particles) / len(union_particles) # similarity = len(shared_particles) / len(union_particles)
similarity = len(shared_particles) / (
len(halo_particle_ids) + len(particle_ids_in_comp_halo) - len(shared_particles))
# assert similarity_orig == similarity
# print(shared_size) # print(shared_size)
# if not similarity: # if not similarity:
# continue # continue
@ -289,7 +292,7 @@ def compare_halo_resolutions(
pl.show() pl.show()
if single: if single:
break break
counters.num_matches=len(compared_halos) counters.num_matches = len(compared_halos)
df = pd.concat(compared_halos, axis=1).T df = pd.concat(compared_halos, axis=1).T
print(df) print(df)
print(f"saving to {outfile}") print(f"saving to {outfile}")