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

fmt: formatting

Browse Source
This commit is contained in:
Deepak Mallubhotla 2023-07-26 20:21:53 -05:00
parent 598dad1e6d
commit 4726ccfb8c
Signed by: deepak
GPG Key ID: BEBAEBF28083E022
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
deepdog/bayes_run_with_ss.py
View File

@ -200,7 +200,9 @@ class BayesRunWithSubspaceSimulation:
for (name, result) in zip(self.model_names, results):
if result.over_target_likelihood is None:
clamped_likelihood = result.probs_list[-1][0] / CLAMPING_FACTOR
_logger.warning(f"got a none result, clamping to {clamped_likelihood}")
_logger.warning(
f"got a none result, clamping to {clamped_likelihood}"
)
else:
clamped_likelihood = result.over_target_likelihood
likelihoods.append(clamped_likelihood)

14
deepdog/subset_simulation/subset_simulation_impl.py
View File

@ -127,8 +127,11 @@ class SubsetSimulation:
next_dipoles_seed_dipoles = numpy.array([n[1] for n in next_seeds])
for n in range(self.model.n):
_logger.info(f"{next_dipoles_seed_dipoles[:, n].shape}")
numpy.savetxt(f"generation_{self.n_c}_{self.n_s}_{i}_dipole_{n}.csv", next_dipoles_seed_dipoles[:, n], delimiter=",")
numpy.savetxt(
f"generation_{self.n_c}_{self.n_s}_{i}_dipole_{n}.csv",
next_dipoles_seed_dipoles[:, n],
delimiter=",",
)
if self.keep_probs_list:
for cost_index, cost_chain in enumerate(all_chains[: -self.n_c]):
@ -187,7 +190,10 @@ class SubsetSimulation:
if self.keep_probs_list:
probs_list.append(
(
((self.n_c * self.n_s - cost_index) / (self.n_c * self.n_s))
(
(self.n_c * self.n_s - cost_index)
/ (self.n_c * self.n_s)
)
/ (self.n_s ** (i)),
cost_chain[0],
i + 1,
@ -234,7 +240,7 @@ class SubsetSimulation:
# _logger.info(f"\t{prob}: {prob_cost}")
probs_list.sort(key=lambda c: c[0], reverse=True)
min_likelihood = ((1) / (self.n_c * self.n_s))/ (self.n_s ** (self.m_max + 1))
min_likelihood = ((1) / (self.n_c * self.n_s)) / (self.n_s ** (self.m_max + 1))
result = SubsetSimulationResult(
probs_list=probs_list,