search

promised-ai / lace

A probabalistic ML tool for science
Other
107 stars 8 forks source link

feat(pylace): Added `lace.StateTransition.common_transition_sets` for… #150

Closed schmidmt closed 1 year ago

schmidmt commented 1 year ago

… commonly used transitions.

This adds a helper set of commonly used transitions to the StateTransition class. For sams, this returns:

>>> import lace
>>> lace.StateTransition.common_transition_sets()["sams"]
[ViewAlphas,
 ColumnAssignment(Sams),
 ViewAlphas,
 ColumnAssignment(Sams),
 ViewAlphas,
 ComponentParams,
 ColumnAssignment(Sams),
 ComponentParams,
 ColumnAssignment(Gibbs),
 ColumnAssignment(Slice),
 ViewAlphas,
 FeaturePriors]