Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Adding online learning to lds model #10

Merged
merged 4 commits into from
May 22, 2024
Merged

Adding online learning to lds model #10

merged 4 commits into from
May 22, 2024

Conversation

ncguilbeault
Copy link
Collaborator

This PR adds online learning capabilities to the existing linear dynamical systems model for kinematics.

Changes to main.py

Several modifications were made to the main.py code. The KalmanFilterKinematics class now contains several new attributes and functions. Critically, the optimize function calls the lds.learning.scipy_optimize_SS_tracking_diagV0 function from the lds package to perform optimisation of the selected parameters. The run_optimization function batches the observations and calls the optimize function. The run_optimization_async function performs the same way, except it creates a separate thread to run optimization and sets a callback function called on_completion to report back when optimization has finished. There are additional functions to check when optimization is finished and to reset the optimization loop.

Include workflows added

Two additional IncludeWorkflows were added. The RunOptimizationAsync workflow allows the model to perform asynchronous optimization of selected model parameters given mini batches of data. The workflow will produce a single output once the optimization procedure has started. Therefore, to check when the optimization has finished, the CheckOptimizationFinished workflow tests on every input whether the model is currently performing optimization and emits a value when the optimization procedure has completed.

@glopesdev glopesdev self-requested a review May 17, 2024 10:15
@glopesdev glopesdev added the feature New planned feature label May 17, 2024
@glopesdev glopesdev merged commit ae92206 into main May 22, 2024
2 checks passed
@glopesdev glopesdev deleted the lds-learning branch May 22, 2024 13:56
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature New planned feature
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants