diff options
| author | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-05 09:30:32 -0400 |
|---|---|---|
| committer | Alex Auvolat <alex.auvolat@ens.fr> | 2015-05-05 09:30:32 -0400 |
| commit | 95b565afb7e1c2a6eb23ca9f7c13cd6efaf55a39 (patch) | |
| tree | 2f40fdf2e80c00ca41c5fb55b597867d2ac9b26b /train.py | |
| parent | 1556e9087f7e49bd75c8e236d2d3fb4fd936dc40 (diff) | |
| download | taxi-95b565afb7e1c2a6eb23ca9f7c13cd6efaf55a39.tar.gz taxi-95b565afb7e1c2a6eb23ca9f7c13cd6efaf55a39.zip | |
New config (added a hidden layer), small changes to train.py
Diffstat (limited to 'train.py')
| -rw-r--r-- | train.py | 4 |
1 files changed, 2 insertions, 2 deletions
@@ -110,7 +110,7 @@ def main(): # Checkpoint('model.pkl', every_n_batches=100), Dump('model_data/' + model_name, every_n_batches=1000), LoadFromDump('model_data/' + model_name), - FinishAfter(after_epoch=5) + FinishAfter(after_epoch=10), ] main_loop = MainLoop( @@ -124,7 +124,7 @@ def main(): # Produce an output on the test data test_stream = setup_test_stream() - outfile = open("test-output.csv", "w") + outfile = open("test-output-%s.csv" % model_name, "w") outcsv = csv.writer(outfile) outcsv.writerow(["TRIP_ID", "LATITUDE", "LONGITUDE"]) for out in apply_model.Apply(outputs=outputs, stream=test_stream, return_vars=['trip_id', 'outputs']): |