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.