fix: define only one shape-para in Gamma model

abess-team · Sep 3, 2023 · d039462 · d039462
1 parent e2ffe0d
commit d039462
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/R-package/R/generate.data.R b/R-package/R/generate.data.R
@@ -393,7 +393,7 @@ generate.data <- function(n,
  # set coef_0 as + abs(min(eta)) + 1
  eta <- eta + abs(min(eta)) + 10
  # set the shape parameter of gamma uniformly in [0.1, 100.1]
- shape_para <- 100 * runif(n) + 0.1
+ shape_para <- rep(100 * runif(1) + 0.1, n)
  y <-
  stats::rgamma(n, shape = shape_para, rate = shape_para * eta)
  }

diff --git a/python/abess/datasets.py b/python/abess/datasets.py
@@ -286,7 +286,7 @@ def __init__(self, n, p, k, family, rho=0, corr_type="const", sigma=1,
  eta = eta - np.abs(np.max(eta)) - 10
  eta = -1 / eta
  # set the shape para of gamma uniformly in [0.1,100.1]
- shape_para = 100 * np.random.uniform(0, 1, n) + 0.1
+ shape_para = 100 * np.random.uniform(0, 1) + 0.1
  y = np.random.gamma(
  shape=shape_para,
  scale=eta / shape_para,