bluegraph.downstream package

Data structures

class bluegraph.downstream.data_structures.ElementClassifier(model, feature_vector_prop=None, feature_props=None, **kwargs)

Interface for graph element classification models.

It wraps a predictive classification model provided by the user and a set of configs that allow the user to fit the model and make predictions on the input PGFrames. Its main goal is to hide the details on converting element (node or edge) properties into data tables that can be provided to the predictive model.

fit(pgframe, train_elements=None, labels=None, label_prop=None, **kwargs)

Fit the classifier.

predict(pgframe, predict_elements=None)

Run prediction on the input graph.


Node classification

Edge prediction