mirror of
https://github.com/Findus23/halo_comparison.git
synced 2024-09-19 16:03:50 +02:00
speed up similarity calculation
This commit is contained in:
parent
22c537aeec
commit
15b43bb7c1
1 changed files with 6 additions and 3 deletions
|
@ -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}")
|
||||||
|
|
Loading…
Reference in a new issue