taxi

Winning entry to the Kaggle taxi competition
git clone https://esimon.eu/repos/taxi.git
Log | Files | Refs | README

time_mlp_1.py (570B)

      1 from blocks.initialization import IsotropicGaussian, Constant
      2 
      3 import data
      4 from model.time_mlp import Model, Stream
      5 
      6 
      7 n_begin_end_pts = 5     # how many points we consider at the beginning and end of the known trajectory
      8 
      9 dim_embeddings = [
     10 ]
     11 
     12 dim_input = n_begin_end_pts * 2 * 2 + sum(x for (_, _, x) in dim_embeddings)
     13 dim_hidden = [200]
     14 dim_output = 1
     15 
     16 embed_weights_init = IsotropicGaussian(0.001)
     17 mlp_weights_init = IsotropicGaussian(0.01)
     18 mlp_biases_init = Constant(0.001)
     19 
     20 exp_base = 1.5
     21 
     22 learning_rate = 0.00001
     23 momentum = 0.99
     24 batch_size = 32
     25 
     26 max_splits = 100