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.