Skip to content

Instantly share code, notes, and snippets.

@vikeshsingh37

vikeshsingh37/snorkel_label_model.py

Created April 5, 2020 14:32
Show Gist options
  • Save vikeshsingh37/ed4cc976d5c3401279567a2ea76ba115 to your computer and use it in GitHub Desktop.
Save vikeshsingh37/ed4cc976d5c3401279567a2ea76ba115 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
snorkel_label_model.py
label_model = LabelModel(cardinality=2, verbose=True)
label_model.fit(L_train, n_epochs=500, log_freq=50, seed=123)
df_train["label"] = label_model.predict(L=L_train, tie_break_policy="abstain")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment