physics/deepdog
2
0
Fork 0
Code Issues 2 Pull Requests 9 Projects Releases Wiki Activity

betterparallel #7

Merged
deepak merged 3 commits from betterparallel into master 2022-04-30 16:15:03 +00:00
Conversation 0 Commits 3 Files Changed 1 +25 -15
1 changed files with 25 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 42829c0327 - Show all commits

40
deepdog/alt_bayes_run_simulpairs.py
View File

@ -33,26 +33,35 @@ def get_a_simul_result_using_pairs(input) -> numpy.ndarray:
nonlocal_lows, nonlocal_lows,
nonlocal_highs, nonlocal_highs,
monte_carlo_count, monte_carlo_count,
monte_carlo_cycles,
max_frequency, max_frequency,
seed, seed,
) = input ) = input
rng = numpy.random.default_rng(seed) rng = numpy.random.default_rng(seed)
sample_dipoles = discretisation.get_model().get_n_single_dipoles( local_total = 0
monte_carlo_count, max_frequency, rng_to_use=rng combined_total = 0
) for i in range(monte_carlo_cycles):
local_vals = pdme.util.fast_v_calc.fast_vs_for_dipoles(dot_inputs, sample_dipoles) sample_dipoles = discretisation.get_model().get_n_single_dipoles(
local_matches = pdme.util.fast_v_calc.between(local_vals, local_lows, local_highs) monte_carlo_count, max_frequency, rng_to_use=rng
nonlocal_vals = pdme.util.fast_nonlocal_spectrum.fast_s_nonlocal( )
pair_inputs, sample_dipoles local_vals = pdme.util.fast_v_calc.fast_vs_for_dipoles(
) dot_inputs, sample_dipoles
nonlocal_matches = pdme.util.fast_v_calc.between( )
nonlocal_vals, nonlocal_lows, nonlocal_highs local_matches = pdme.util.fast_v_calc.between(
) local_vals, local_lows, local_highs
combined_matches = numpy.logical_and(local_matches, nonlocal_matches) )
return numpy.array( nonlocal_vals = pdme.util.fast_nonlocal_spectrum.fast_s_nonlocal(
[numpy.count_nonzero(local_matches), numpy.count_nonzero(combined_matches)] pair_inputs, sample_dipoles
) )
nonlocal_matches = pdme.util.fast_v_calc.between(
nonlocal_vals, nonlocal_lows, nonlocal_highs
)
combined_matches = numpy.logical_and(local_matches, nonlocal_matches)
local_total += numpy.count_nonzero(local_matches)
combined_total += numpy.count_nonzero(combined_matches)
return numpy.array([local_total, combined_total])
class AltBayesRunSimulPairs: class AltBayesRunSimulPairs:
@ -238,6 +247,7 @@ class AltBayesRunSimulPairs:
pair_lows, pair_lows,
pair_highs, pair_highs,
self.monte_carlo_count, self.monte_carlo_count,
self.monte_carlo_cycles,
self.max_frequency, self.max_frequency,
seed, seed,
) )