Skip to content

Conversation

@GabrielleCasimiro
Copy link
Contributor

Add Covariate Model Structure

This PR introduces the full implementation of the covariate model, following the standard architecture:

  • CovariateTimeReparametrizedModel inherits from McmcSaemCompatibleModel
  • CovariateRiemanianManifoldModel inherits from CovariateTimeReparametrizedModel
  • CovariateLogisticModel inherits from:
    • CovariateLogisticInitializationMixin
    • CovariateRiemanianManifoldModel

Notes

  • The model runs, and the population parameters g and v0 are properly updated during the iterations.
  • Known issue: the tau parameters are not updated during training yet. This behavior needs further investigation.

lea-aguilhon and others added 4 commits July 4, 2025 15:49
* add docstrings

* changes

* Apply suggestions from code review

Co-authored-by: caglayantuna <[email protected]>

---------

Co-authored-by: caglayantuna <[email protected]>
* changes

* modifications after suggestions
@caglayantuna
Copy link
Contributor

Hello @GabrielleCasimiro , can you check if phi_tau is updated during the iterations? The issue should be solved here. Am I right?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants