bluegraph.core.embed package¶
Reference page for the bluegraph.core.embed package.
Graph embedding¶
- class bluegraph.core.embed.embedders.Embedder¶
Embedder inferface for EmbeddingPipeline.
- abstract fit_model(data, **kwargs)¶
Train specified model on the provided data.
- abstract info()¶
Get dictionary with the info.
- abstract predict_embeddings(data=None, **kwargs)¶
Predict embeddings of out-sample elements.
- class bluegraph.core.embed.embedders.GraphElementEmbedder(model_name, directed=True, include_type=False, feature_props=None, feature_vector_prop=None, edge_weight=None, **model_params)¶
Abstract class for a node/edge embedder.
- exception FittingException¶
Exception class for fitting errors.
- exception FittingWarning¶
Exception class for fitting errors.
- exception InvalidModelException¶
Exception class for invalid model names.
- exception PredictionException¶
Exception class for fitting errors.
- fit_model(pgframe)¶
Fit the embedding model.
- info()¶
Get dictionary with the info.
- static load(path)¶
Load a dumped embedder.
- predict_embeddings(pgframe, nodes=None)¶
Predict embeddings of out-sample elements.
- print_info()¶
Print embedder info.
- save(path, compress=False, save_graph=False)¶
Save the embedder.
- class bluegraph.core.embed.embedders.GraphEmbedder¶
Abstract class for a graph embedder.