Skip to content

Logan-Lin/ST-LSTM_PyTorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

Spatial-Temporal LSTM network proposed in Kong D, Wu F. HST-LSTM: A Hierarchical Spatial-Temporal Long-Short Term Memory Network for Location Prediction[C]//IJCAI. 2018: 2341-2347. Implemented with PyTorch.

  • Core implementation is in stlstm.py - STLSTMCell.
  • An example is presented in stlstm_nextloc.py. Some implementation is modified to fit into my task, but the calculation of slots and process of linear interpolation is general.

About

ST-LSTM network implemented using PyTorch.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages