search

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

Consolidate task encoder #40

Closed cnellington closed 2 years ago

cnellington commented 2 years ago

Make the beta and mu task encoders into a single task encoder with a (2 x archetype_dim) output where ones row is the beta encoding and the other is the mu encoding. This gives a single subtype per model.