PythonSetFilterer

class PythonSetFilterer(mapped_triples: Tensor)[source]

Bases: Filterer

A filterer using Python sets for filtering.

This filterer is expected to be rather slow due to the conversion from torch long tensors to Python tuples. It can still serve as a baseline for performance comparison.

Initialize the filterer.

Parameters:: mapped_triples (Tensor) – The ID-based triples.

Methods Summary

contains(batch)

Check whether a triple is contained.

Methods Documentation

contains(batch: Tensor) → Tensor[source]

Check whether a triple is contained.

Supports batching.

Parameters:: batch (Tensor) – shape (batch_size, 3) The batch of triples.
Returns:: shape: (batch_size,) Whether the triples are contained in the training triples.
Return type:: Tensor