search

ramanathanlab / mdlearn

Machine learning for molecular dynamics
MIT License
13 stars 7 forks source link

LSTM model for feature vector time series inputs #34

Closed braceal closed 3 years ago

braceal commented 3 years ago

Description

An LSTM Module, Trainer class, and Dataset which allows prediction of feature vectors given a window of previous feature vectors.

braceal commented 3 years ago

41 Addresses this. The model code is implemented with a Trainer. However, the model still does not produce good results. Needs more work. There is a warning about this in the public documentation.