gigl.src.inference.v1.InferencerV1#

class gigl.src.inference.v1.gnn_inferencer.InferencerV1(bq_gcp_project: str)#

Bases: object

****** WILL BE DEPRECATED ****** Note this class will get deprecated in the future without notice Use python/gigl/src/inference/inferencer.py instead ****** WILL BE DEPRECATED ****** GiGL Component that runs inference of a trained model on samples generated by the Subgraph Sampler component and outputs embedding and/or prediction assets.

Methods

__init__

generate_inferencer_instance

run

write_from_gcs_to_bq

Writes embeddings or predictions from gcs folder to bq table with specified schema Args: schema (Optional[Dict[str, List[Dict[str, str]]]): BQ Table schema for embeddings or predictions from inference output gcs_uri (GcsUri): GCS Folder for embeddings or predictions from inference output bq_table_uri (str): Path to the table for embeddings or predictions output

__init__(bq_gcp_project: str)#

__weakref__#: list of weak references to the object (if defined)

write_from_gcs_to_bq(schema: Dict[str, List[Dict[str, str]]], gcs_uri: GcsUri, bq_table_uri: str) → None#: Writes embeddings or predictions from gcs folder to bq table with specified schema Args:

schema (Optional[Dict[str, List[Dict[str, str]]]): BQ Table schema for embeddings or predictions from inference output gcs_uri (GcsUri): GCS Folder for embeddings or predictions from inference output bq_table_uri (str): Path to the table for embeddings or predictions output

`__init__`
`generate_inferencer_instance`
`run`
`write_from_gcs_to_bq`	Writes embeddings or predictions from gcs folder to bq table with specified schema Args: schema (Optional[Dict[str, List[Dict[str, str]]]): BQ Table schema for embeddings or predictions from inference output gcs_uri (GcsUri): GCS Folder for embeddings or predictions from inference output bq_table_uri (str): Path to the table for embeddings or predictions output