gigl.experimental.knowledge_graph_embedding.lib.constants.bq#

Functions#

get_dst_node_embedding_table_for_edge_type(...[, ...])

Returns the BigQuery table for the embeddings of a specific destination node type.

get_src_node_embedding_table_for_edge_type(...[, ...])

Returns the BigQuery table for the embeddings of a specific source node type.

Module Contents#

gigl.experimental.knowledge_graph_embedding.lib.constants.bq.get_dst_node_embedding_table_for_edge_type(applied_task_identifier, edge_type, is_enumerated=False)[source]#

Returns the BigQuery table for the embeddings of a specific destination node type.

Parameters:

  • applied_task_identifier (gigl.src.common.types.AppliedTaskIdentifier) – The identifier of the applied task.

  • edge_type (gigl.src.common.types.graph_data.EdgeType) – The edge type for which to get the embedding table.

  • is_enumerated (bool) – Whether the embeddings are enumerated (default is False).

Returns:

The BigQuery table name for the embeddings of the specified source node type.

Return type:

str

gigl.experimental.knowledge_graph_embedding.lib.constants.bq.get_src_node_embedding_table_for_edge_type(applied_task_identifier, edge_type, is_enumerated=False)[source]#

Returns the BigQuery table for the embeddings of a specific source node type.

Parameters:

  • applied_task_identifier (gigl.src.common.types.AppliedTaskIdentifier) – The identifier of the applied task.

  • edge_type (gigl.src.common.types.graph_data.EdgeType) – The edge type for which to get the embedding table.

  • is_enumerated (bool) – Whether the embeddings are enumerated (default is False).

Returns:

The BigQuery table name for the embeddings of the specified source node type.

Return type:

str