fdl.ImageEmbedding

Represents custom features derived from image embeddings.

Input ParameterTypeDefaultDescription
typeCustomFeatureTypeCustomFeatureType.FROM_IMAGE_EMBEDDINGIndicates this feature is derived from an image embedding.
source_columnstrRequiredURL where image data is stored
columnstrRequiredSpecifies the column name where embeddings corresponding to source_col are stored.
n_clustersOptional[int]5The number of clusters
centroidsOptional[List]Centroids of the clusters in the embedded space. Number of centroids equal to n_clustersCentroids of the clusters in the embedded space. Number of centroids equal to n_clusters
image_embedding_feature = fdl.ImageEmbedding(
    name='image_feature',
    source_column='image_url',
    column='image_embedding',
)