flair.datasets.relation_extraction.RE_ENGLISH_DRUGPROT

class flair.datasets.relation_extraction.RE_ENGLISH_DRUGPROT(base_path=None, in_memory=True, sentence_splitter=<flair.splitter.SegtokSentenceSplitter object>, **corpusargs)View on GitHub

Bases: ColumnCorpus

__init__(base_path=None, in_memory=True, sentence_splitter=<flair.splitter.SegtokSentenceSplitter object>, **corpusargs)View on GitHub

Initialize the DrugProt corpus.

Biocreative VII Track 1 from https://zenodo.org/record/5119892#.YSdSaVuxU5k/ on drug and chemical-protein interactions.

Methods

__init__([base_path, in_memory, ...])	Initialize the DrugProt corpus.
add_label_noise(label_type, labels[, ...])	Generates uniform label noise distribution in the chosen dataset split.
char_spans_to_token_spans(char_spans, ...)
downsample([percentage, downsample_train, ...])	Randomly downsample the corpus to the given percentage (by removing data points).
drugprot_document_to_tokenlists(pmid, ...)
extract_and_convert_to_conllu(data_file, ...)
filter_empty_sentences()	A method that filters all sentences consisting of 0 tokens.
filter_long_sentences(max_charlength)	A method that filters all sentences for which the plain text is longer than a specified number of characters.
get_all_sentences()	Returns all sentences (spanning all three splits) in the Corpus.
get_label_distribution()	Counts occurrences of each label in the corpus and returns them as a dictionary object.
has_overlap(a, b)
make_label_dictionary(label_type[, ...])	Creates a dictionary of all labels assigned to the sentences in the corpus.
make_tag_dictionary(tag_type)	Create a tag dictionary of a given label type.
make_vocab_dictionary([max_tokens, min_freq])	Creates a Dictionary of all tokens contained in the corpus.
obtain_statistics([label_type, pretty_print])	Print statistics about the corpus, including the length of the sentences and the labels in the corpus.

Attributes

dev	The dev split as a torch.utils.data.Dataset object.
test	The test split as a torch.utils.data.Dataset object.
train	The training split as a torch.utils.data.Dataset object.

extract_and_convert_to_conllu(data_file, data_folder)View on GitHub

char_spans_to_token_spans(char_spans, token_offsets)View on GitHub

has_overlap(a, b)View on GitHub

drugprot_document_to_tokenlists(pmid, title_sentences, abstract_sentences, abstract_offset, entities, relations)View on GitHub

Return type:

list[TokenList]