Classes

BrGraph

alias of Bering_Graph

models.GCN(n_features, n_classes[, ...])

Node classification Model with Graph Convolutional Networks (GCN) and Multilayer Perceptron (MLP).

models.BaselineMLP(n_features, n_classes[, ...])

Node classification Model with Multilayer Perceptron (MLP) without using graph models.

models.ImageEncoder(image_dims[, ...])

Convolutional neural network to learn representation from staining images of different sizes.

models.EdgeClf(n_node_latent_features, image)

Edge classifier model which learns node classification embedding, image embedding and distance kernel

training.EarlyStopper([patience, min_delta])

Early Stop implementation for loss values

training.TrainerNode([model, lr, ...])

Trainer for node classification model.

training.TrainerEdge(model, nodeclf_model, ...)

Trainer for edge classification model.