Merge pull request #1298 from perib/random_state_fix

nickotto · web-flow · commit 5277f10540f0 · 2023-05-18T14:01:56.000-07:00
Random state fix
diff --git a/tpot/base.py b/tpot/base.py
@@ -1420,7 +1420,7 @@ def _compile_to_sklearn(self, expr):
         )
         sklearn_pipeline = eval(sklearn_pipeline_str, self.operators_context)
         sklearn_pipeline.memory = self._memory
-        if self.random_state:
+        if self.random_state is not None:
             # Fix random state when the operator allows
             set_param_recursive(
                 sklearn_pipeline.steps, "random_state", self.random_state

Original file line number	Diff line number	Diff line change
`@@ -1420,7 +1420,7 @@ def _compile_to_sklearn(self, expr):`
`1420`	`1420`	`)`
`1421`	`1421`	`sklearn_pipeline = eval(sklearn_pipeline_str, self.operators_context)`
`1422`	`1422`	`sklearn_pipeline.memory = self._memory`
`1423`		`- if self.random_state:`
	`1423`	`+ if self.random_state is not None:`
`1424`	`1424`	`# Fix random state when the operator allows`
`1425`	`1425`	`set_param_recursive(`
`1426`	`1426`	`sklearn_pipeline.steps, "random_state", self.random_state`