cnellington / Contextualized

An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.
http://contextualized.ml/
GNU General Public License v3.0
65 stars 9 forks source link

DAGs Cleanup #129

Closed cnellington closed 1 year ago

cnellington commented 1 year ago

Cleaned the dags module to use similar patterns as contextualized.regression. Setting up for a contextualized.graphs module including correlation and Markov networks.

Removed datamodules, baselines, and callbacks files from the dags module (check the commits before this PR if there's any need for these in the future).

Note: This NOTMAD version converges to the true parameters in tests, but requires more epochs (50 vs. 5). Possibly because dynamic alpha/rho was being called twice per epoch in previous tests? Will be looking into this, but making this PR so that #128 can continue.