We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
Categorical.rvs
1 parent d9b8ec1 commit d4a9f14Copy full SHA for d4a9f14
preliz/distributions/categorical.py
@@ -127,7 +127,8 @@ def kurtosis(self):
127
128
def rvs(self, size=None, random_state=None):
129
random_state = np.random.default_rng(random_state)
130
- return random_state.choice(self.p, size)
+ support_array = np.arange(self.support[0], self.support[1] + 1)
131
+ return random_state.choice(support_array, size, p=self.p)
132
133
def _fit_mle(self, sample):
134
optimize_ml(self, sample)
0 commit comments