Towards Unified Sequence Representation III: Add SequenceEmbeddingModel for SequenceProcessorV2 fields

docs/api/models.rst

@@ -247,6 +247,7 @@ API Reference
     models/pyhealth.models.CorGAN
     models/pyhealth.models.Synthea
     models/pyhealth.models.SDOH
+    models/pyhealth.models.SequenceEmbeddingModel
     models/pyhealth.models.VisionEmbeddingModel
     models/pyhealth.models.TextEmbedding
     models/pyhealth.models.BIOT

docs/api/models/pyhealth.models.SequenceEmbeddingModel.rst

@@ -0,0 +1,18 @@
+pyhealth.models.SequenceEmbeddingModel
+======================================
+
+Embeddings for the packed records of ``sequence_v2``
+(:class:`~pyhealth.processors.SequenceProcessorV2`) fields. Each level's
+records get one embedding per token namespace and a projection of their
+values, concatenated; each level keeps its offsets and times.
+
+.. autoclass:: pyhealth.models.SequenceEmbeddingModel
+    :members:
+    :undoc-members:
+    :show-inheritance:
+
+.. autoclass:: pyhealth.models.LevelEmbedding
+    :members:
+
+.. autoclass:: pyhealth.models.PretrainedEmbedding
+    :members:

pyhealth/models/__init__.py

@@ -8,6 +8,11 @@
 from .contrawr import ContraWR, ResBlock2D
 from .deepr import Deepr, DeeprLayer
 from .embedding import EmbeddingModel
+from .sequence_embedding import (
+    LevelEmbedding as LevelEmbedding,
+    PretrainedEmbedding as PretrainedEmbedding,
+    SequenceEmbeddingModel as SequenceEmbeddingModel,
+)
 from .gamenet import GAMENet, GAMENetLayer
 from .jamba_ehr import JambaEHR, JambaLayer
 from .logistic_regression import LogisticRegression