Skip to content

Commit

Permalink
Merge pull request EpistasisLab#330 from Yatoom/patch-1
Browse files Browse the repository at this point in the history
Fixed small problem when random_state = 0
  • Loading branch information
rhiever authored Dec 19, 2016
2 parents db30ccf + 116d4d0 commit c71af7a
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tpot/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ def __init__(self, population_size=100, generations=100,
How many minutes TPOT has to optimize a single pipeline.
Setting this parameter to higher values will allow TPOT to explore more complex
pipelines but will also allow TPOT to run longer.
random_state: int (default: 0)
random_state: int (default: None)
The random number generator seed for TPOT. Use this to make sure
that TPOT will give you the same results each time you run it
against the same data set with that seed.
Expand Down Expand Up @@ -286,7 +286,7 @@ def fit(self, features, classes):
features = features.astype(np.float64)

# Set the seed for the GP run
if self.random_state:
if self.random_state is not None:
random.seed(self.random_state)
np.random.seed(self.random_state)

Expand Down

0 comments on commit c71af7a

Please sign in to comment.