YuliaRubanova / latent_ode

Code for "Latent ODEs for Irregularly-Sampled Time Series" paper
MIT License
522 stars 126 forks source link

Trained model, now how do I apply to forecast on another unseen time-series? #13

Open nina-thigpen opened 3 years ago

nina-thigpen commented 3 years ago

So I load this checkpoint path from the trained latent ODE model: ckpt_path = '/experiments/experiment_11387.ckpt' checkpt = torch.load(ckpt_path)

If I want to feed a new unseen time-series into the model and visualize the forecast, should I use LatentODE().get_reconstruct? If so, how do I initialize the model from the weights from the checkpoint path (i.e. the model I just trained)?

I'm trying to use the Visualize functions but they seem to need a data_dict and I can't find where you make this or what components it should have.