aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAlex Auvolat <alex.auvolat@ens.fr>2015-04-29 18:36:16 -0400
committerAlex Auvolat <alex.auvolat@ens.fr>2015-04-29 18:36:16 -0400
commite28390de61b23882f6e1069d565a2137825c2662 (patch)
tree1e173130733201bcd41f7f64e84e9af150b59be7
parentfbcaf8643a68ef824c72b003c85c4a96e5362bb6 (diff)
downloadtaxi-e28390de61b23882f6e1069d565a2137825c2662.tar.gz
taxi-e28390de61b23882f6e1069d565a2137825c2662.zip
Performance improvement
-rw-r--r--model.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/model.py b/model.py
index c89809f..6a8be9d 100644
--- a/model.py
+++ b/model.py
@@ -150,7 +150,7 @@ def main():
every_n_batches=1000),
Printing(every_n_batches=1000),
# Checkpoint('model.pkl', every_n_batches=100),
- Dump('taxi_model', every_n_batches=100),
+ Dump('taxi_model', every_n_batches=1000),
LoadFromDump('taxi_model'),
]
@@ -160,6 +160,7 @@ def main():
algorithm=algorithm,
extensions=extensions)
main_loop.run()
+ main_loop.profile.report()
# Produce an output on the test data
'''