File tree Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Expand file tree Collapse file tree 1 file changed +7
-7
lines changed Original file line number Diff line number Diff line change @@ -306,13 +306,6 @@ def basic_test(
306306 else :
307307 raise ValueError ("noise model %s not recognized" % self .noise_model )
308308
309- estimator = _TestAccuracyGlmAllEstim (
310- simulator = self .simulator (train_loc = train_loc ),
311- quick_scale = False if train_scale else True ,
312- noise_model = self .noise_model ,
313- sparse = sparse ,
314- init_mode = init_mode
315- )
316309 for algo in algos :
317310 logger .info ("algorithm: %s" % algo )
318311 if algo in ["ADAM" , "RMSPROP" , "GD" ]:
@@ -348,6 +341,13 @@ def basic_test(
348341 glm .pkg_constants .JACOBIAN_MODE = "analytic"
349342 else :
350343 return ValueError ("algo %s not recognized" % algo )
344+ estimator = _TestAccuracyGlmAllEstim (
345+ simulator = self .simulator (train_loc = train_loc ),
346+ quick_scale = False if train_scale else True ,
347+ noise_model = self .noise_model ,
348+ sparse = sparse ,
349+ init_mode = init_mode
350+ )
351351 estimator .estimate (
352352 algo = algo ,
353353 batched = batched ,
You can’t perform that action at this time.
0 commit comments