search

applied-ml-bde / bde

Bayesian Deep Ensembles
https://applied-ml-bde.github.io/bde/
BSD 3-Clause "New" or "Revised" License
2 stars 0 forks source link

Save estimators #30

Closed cklyne closed 1 month ago

cklyne commented 2 months ago

Idea

Implement a save() and a load() method for the estimator classes.

Dependencies

Comments

Since the estimator classes are PyTree serializable, this functionality can probably be easily implemented by using tree_flatten() and tree_unflatten().