Struct BoxEmbedding 

pub struct BoxEmbedding {
    pub min: Vec<f32>,
    pub max: Vec<f32>,
    pub temperature: Option<f32>,
}

Box embedding for geometric coreference resolution.

Uses axis-aligned hyperrectangles to encode logical invariants. This is based on research showing that box embeddings can capture containment relationships better than vector embeddings.

Status: Experimental

This type may change significantly. The current representation uses serde_json::Value as a placeholder; a proper typed representation will be added once the embedding format stabilizes.

References

• Vilnis et al., “Probabilistic Embedding of Knowledge Graphs with Box Lattice Measures”
• Dasgupta et al., “Improving Local Identifiability in Probabilistic Box Embeddings”

Fields

min: Vec<f32>

Minimum corner of the box (lower bounds).

max: Vec<f32>

Maximum corner of the box (upper bounds).

temperature: Option<f32>

Temperature parameter for softbox formulation.

Implementations

impl BoxEmbedding

pub fn new(min: Vec<f32>, max: Vec<f32>) -> Self

Create a new box embedding.

pub fn dim(&self) -> usize

Get the dimensionality of the box.

pub fn is_valid(&self) -> bool

Check if this box is valid (min <= max in all dimensions).

pub fn volume(&self) -> f32

Compute the volume of the box (product of side lengths).

pub fn contains_point(&self, point: &[f32]) -> bool

Check if this box contains a point.

pub fn contains_box(&self, other: &BoxEmbedding) -> bool

Check if this box contains another box.

pub fn intersection_volume(&self, other: &BoxEmbedding) -> f32

Compute intersection volume with another box.

pub fn from_json(value: &Value) -> Option<Self>

Convert from a JSON value (for backwards compatibility with existing data).

pub fn to_json(&self) -> Value

Convert to a JSON value (for backwards compatibility).

Trait Implementations

impl Clone for BoxEmbedding

fn clone(&self) -> BoxEmbedding

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for BoxEmbedding

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for BoxEmbedding

fn default() -> BoxEmbedding

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for BoxEmbedding

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for BoxEmbedding

fn eq(&self, other: &BoxEmbedding) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for BoxEmbedding

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for BoxEmbedding