diff options
| author | Étienne Simon <esimon@esimon.eu> | 2015-04-29 15:40:05 -0400 |
|---|---|---|
| committer | Étienne Simon <esimon@esimon.eu> | 2015-04-29 15:40:51 -0400 |
| commit | f768d3e770216d4227ffd989cf98f1628fc476a3 (patch) | |
| tree | b5620a34eaebac5290b74882018bab16f4658e27 /config/model_0.py | |
| parent | 61e0d47b6c6a570feebb43d474138020b13495aa (diff) | |
| download | taxi-f768d3e770216d4227ffd989cf98f1628fc476a3.tar.gz taxi-f768d3e770216d4227ffd989cf98f1628fc476a3.zip | |
Adapt model to hdf5 dataset. WIP
Diffstat (limited to 'config/model_0.py')
| -rw-r--r-- | config/model_0.py | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/config/model_0.py b/config/model_0.py new file mode 100644 index 0000000..ba04a15 --- /dev/null +++ b/config/model_0.py @@ -0,0 +1,21 @@ +n_dow = 7 # number of division for dayofweek/dayofmonth/hourofday +n_dom = 31 +n_hour = 24 + +n_clients = 57124 #57105 +n_stands = 63 + +n_begin_end_pts = 5 # how many points we consider at the beginning and end of the known trajectory +n_end_pts = 5 + +train_size = 1710670 +n_valid = 1000 + +dim_embed = 50 +dim_input = n_begin_end_pts * 2 * 2 + dim_embed + dim_embed +dim_hidden = [200] +dim_output = 2 + +learning_rate = 0.002 +momentum = 0.9 +batch_size = 32 |