API
===

.. toctree::
    :maxdepth: 2
    :caption: Packages
    :titlesonly:

    langspace/langspace.metrics
    langspace/langspace.models
    langspace/langspace.ops
    langspace/langspace.probe.arithmetic
    langspace/langspace.probe.cluster_vis
    langspace/langspace.probe.disentanglement
    langspace/langspace.probe.interpolation
    langspace/langspace.probe.traversal


Latent probe classes
-------------------------------------------

.. autosummary::
    ~langspace.probe.arithmetic.ArithmeticProbe
    ~langspace.probe.cluster_vis.ClusterVisualizationProbe
    ~langspace.probe.disentanglement.DisentanglementProbe
    ~langspace.probe.interpolation.InterpolationProbe
    ~langspace.probe.traversal.TraversalProbe
    :nosignatures:


Metrics
-------------------------------------------

.. autosummary::
    ~langspace.metrics.disentanglement.DisentanglementMetric
    ~langspace.metrics.interpolation.InterpolationMetric
    :nosignatures:


Operations
-------------------------------------------

.. autosummary::
    ~langspace.ops.arithmetic.ArithmeticOps
    ~langspace.ops.interpolation.InterpolationOps
    ~langspace.ops.traversal.TraversalOps
    :nosignatures:


Utility
-------------------------------------------

.. autosummary::
    ~langspace.models.select_models
    :nosignatures: