Skip to content

Commit

Permalink
Update default params for TPOT example
Browse files Browse the repository at this point in the history
  • Loading branch information
rhiever authored Aug 20, 2016
1 parent 6f63e83 commit 9c8c0e2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@ digits = load_digits()
X_train, X_test, y_train, y_test = train_test_split(digits.data, digits.target,
train_size=0.75, test_size=0.25)

tpot = TPOT(generations=5)
tpot = TPOT(generations=5, population_size=20, verbosity=2)
tpot.fit(X_train, y_train)
print(tpot.score(X_test, y_test))
tpot.export('tpot_mnist_pipeline.py')
Expand Down

0 comments on commit 9c8c0e2

Please sign in to comment.