crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 05:35:49 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adding docstrings to normalize methods

Browse source
This commit is contained in:
Patrick Shriwise 2022-06-19 09:36:31 -05:00
parent f52865854b
commit f46402fd87
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
openmc/stats/univariate.py
View file

@ -142,6 +142,7 @@ class Discrete(Univariate):
return np.random.choice(self.x, n_samples, p=self.p)
def normalize(self):
"""Normalize the probabilities stored on the distribution"""
norm = sum(self.p)
self.p = [val / norm for val in self.p]
@ -892,6 +893,7 @@ class Tabular(Univariate):
return np.cumsum(c)
def normalize(self):
"""Normalize the probabilities stored on the distribution"""
self.p = np.asarray(self.p) / self.cdf().max()
def sample(self, n_samples=1, seed=None):
@ -1115,6 +1117,7 @@ class Mixture(Univariate):
return out
def normalize(self):
"""Normalize the probabilities stored on the distribution"""
norm = sum(self.probability)
self.probability = [val / norm for val in self.probability]