Skip to main content
Version: 25.2

snorkelflow.client.models

Model related functions.

Functions

add_predictions(node, *, model_uid, x_uids)Register predictions generated by models (and optionally, ground truth labels).
get_models(node)Return a list of models in pandas dataframe.
get_predictions(node[, split, model_uid, ...])Get predictions from a trained model.
register_model(node, description, training_set)Register a model with Snorkel Training Data Manager.
register_trained_model(model, node, description)Register a trained model with Snorkel.
train_custom_model(node, custom_cls, ...[, ...])Train a custom machine learning model.
train_model(node, feature_fields, ...[, ...])Train a machine learning model.