Skip to main content
Version: 0.91

templates.EmbeddingNNTemplateSchema

class templates.EmbeddingNNTemplateSchema(*args, **kwargs)

Template schema for EmbeddingsNearestNeighbor

Parameters:
  • class_embeddings_path (str) – Path to a file containing class embeddings

  • feature_field (str) – Name of the field containing the feature / embeddings

  • min_distance (float) – Minimum distance to be considered a match

static artifact_config_keys()

The names of the fields in the template schema pydantic model that represent paths to files or directories that contain data which back the LF template. For example, paths to a dictionary on disk that’s used within an LF.

Return type:

List[str]