em.py 文件源码

def gen(self, N, trials, normal_p_range, anomaly_p_range, anomaly_scale = 1.0):
    self.N = N
    self.trials = trials

    self.gens = [
      ?ompound_distribution(
        stats.uniform(loc=normal_p_range[0], scale=normal_p_range[1] - normal_p_range[0]),
        lambda a: stats.gamma(a = a, scale = 1.0)
      ),

      ?ompound_distribution(
        stats.uniform(loc=anomaly_p_range[0], scale=anomaly_p_range[1] - anomaly_p_range[0]),
        lambda a: stats.gamma(a = a, scale = anomaly_scale)
      )
    ]

    self.priors = np.array([0.9, 0.1])

    self.cats, self.params, self.X = compound_rvs(self.gens, self.priors, self.N, self.trials)