diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-07-24 10:40:34 -0400 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-07-24 10:40:34 -0400 |
| commit | ac49e3cb892e0278ea1d52afdc314322000fae27 (patch) | |
| tree | c52acb2319cc25e13ad7d718e0e31b500e920aa5 | |
| parent | 1e64a442e78b5e471b2f573295bd9a747b7c6c3f (diff) | |
| download | taxi-ac49e3cb892e0278ea1d52afdc314322000fae27.tar.gz taxi-ac49e3cb892e0278ea1d52afdc314322000fae27.zip | |
Fix test only when valid improve
| -rw-r--r-- | ext_test.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ext_test.py b/ext_test.py index 092f39c..5bb4520 100644 --- a/ext_test.py +++ b/ext_test.py @@ -55,12 +55,14 @@ class RunOnTest(SimpleExtension): return if output_dvc: + self.best_dvc = dvc dest_outname = 'test-dest-%s-it%09d-cost%.3f.csv' % (self.model_name, iter_no, dvc) dest_outfile = open(os.path.join('output', dest_outname), 'w') dest_outcsv = csv.writer(dest_outfile) dest_outcsv.writerow(["TRIP_ID", "LATITUDE", "LONGITUDE"]) logger.info("Generating output for test set: %s" % dest_outname) if output_tvc: + self.best_tvc = tvc time_outname = 'test-time-%s-it%09d-cost%.3f.csv' % (self.model_name, iter_no, tvc) time_outfile = open(os.path.join('output', time_outname), 'w') time_outcsv = csv.writer(time_outfile) |