ml_drought icon indicating copy to clipboard operation
ml_drought copied to clipboard

[WIP] Add an embedding layer for the months

Open gabrieltseng opened this issue 6 years ago • 1 comments

As per Yasser's comments, replaces the one hot encoding with an embedding.

~Currently, the size of the embedding is fixed at 12, but if this works well this could become an additional argument.~ The size of the embedding is an additional argument - if None is passed, the prediction month isn't passed to the model.

Also, removes the EALSTM cell written using nn.Modules, since the original one works fine with shap.

gabrieltseng avatar Aug 27 '19 21:08 gabrieltseng

@gabrieltseng is this still good to merge?

tommylees112 avatar Aug 11 '20 09:08 tommylees112