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

feat: clamps results now
Some checks failed
gitea-physics/deepdog/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Deepak Mallubhotla
2023-07-24 10:24:23 -05:00
parent f775ed34c6
commit 9bb8fc50fe
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
deepdog/bayes_run_with_ss.py
View File

@@ -20,6 +20,8 @@ CHUNKSIZE = 50
DotInput = Tuple[numpy.typing.ArrayLike, float]
CLAMPING_FACTOR = 10
_logger = logging.getLogger(__name__)
@@ -192,11 +194,12 @@ class BayesRunWithSubspaceSimulation:
likelihoods: List[float] = []
for (name, result) in zip(self.model_names, results):
clamped_likelihood = result.over_target_likelihood
if result.over_target_likelihood is None:
_logger.error("got a none result, bye")
return
likelihoods.append(result.over_target_likelihood)
row[f"{name}_likelihood"] = result.over_target_likelihood
clamped_likelihood = result.probs_list[-1][0] / CLAMPING_FACTOR
_logger.warning(f"got a none result, clamping to {}")
likelihoods.append(clamped_likelihood)
row[f"{name}_likelihood"] = result.clamped
success_weight = sum(
[