langspace.probe.sts package

Module contents

class langspace.probe.sts.STSProbe(model: LangVAE, data: Iterable[str | Sentence], sample_size: int)

Bases: LatentSpaceProbe

Class for probing the semantic textual similarity in the latent space of a language VAE.

report() → DataFrame

Generate a report from the probe.

Returns:

The generated report.

Return type:

DataFrame