/LaserSTM

Modeling lasers with Machine Learning

Primary LanguageJupyter NotebookCreative Commons Zero v1.0 UniversalCC0-1.0

NeurIPS2023

Data Preprocessing, LSTM Model, and Analysis for something.

TODO:

  1. Unify single passes and training functions if they're the same (seems to be at least in case of LSTM and NFO, I don't see why we can't get a general training function for all models)
  2. Separate dataset and data loader for the CustomSequence (will eff up training code but we'll fix it)
  3. Fix the transformer model (needs proper testing)

This is a continuation of [https://github.com/jhirschm/NeurIPS2023]