A stochastic Koopman model is formulated as follows:
A moving horizon estimation with automatic generated weighting matrices is designed. The optimization problem is in the following form:
with the objective function to be
and the stage cost
The neural networks are built and trained using PyTorch. Two types of loss functions are used for training:
- Data-driven loss terms
- Physics informed loss terms
The model structure is shown as follows:
