Skip to content

Commit ec67a92

Browse files
authored
Merge pull request #31 from LSSTDESC/issue/29/rename
change trainfrac to train_frac
2 parents fdbbe5a + 6db00b4 commit ec67a92

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • src/rail/estimation/algos

src/rail/estimation/algos/gpz.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ class GPzInformer(CatInformer):
5656
config_options.update(
5757
nondetect_val=SHARED_PARAMS,
5858
mag_limits=SHARED_PARAMS,
59-
trainfrac=Param(
59+
train_frac=Param(
6060
float,
6161
0.75,
6262
msg="fraction of training data used to make tree, rest used to set best sigma",
@@ -136,7 +136,7 @@ def run(self):
136136
# need permutation mask to define training vs validation
137137
ngal = input_array.shape[0]
138138
print(f"ngal: {ngal}")
139-
ntrain = int(ngal * self.config.trainfrac)
139+
ntrain = int(ngal * self.config.train_frac)
140140
randvec = np.random.permutation(ngal)
141141
train_mask = np.zeros(ngal, dtype=bool)
142142
val_mask = np.zeros(ngal, dtype=bool)

0 commit comments

Comments
 (0)