Skip to content

Instantly share code, notes, and snippets.

@dvsrepo

dvsrepo/rubrix_ner.py

Last active June 1, 2021 21:39
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dvsrepo/b11d2b4046af8ef7127bc7661bd9f825 to your computer and use it in GitHub Desktop.
Save dvsrepo/b11d2b4046af8ef7127bc7661bd9f825 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
rubrix_ner.py
text = "I love the song Computer Love from Kraftwerk"
record = rb.TokenClassificationRecord(
text=text,
tokens=[t for t in text.split(' ')],
prediction=[("SONG", 16, 29), ("BAND", 35, 44)],
prediction_agent="my_ner_model_v1"
)
rb.log(record, name="ner_bands_dataset")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment