MCPcopy Create free account

hub / github.com/amazon-science/ReFinED / types & classes

Types & classes58 in github.com/amazon-science/ReFinED

↓ 29 callersClassDate
src/refined/utilities/numeric_handling/date_utils.py:12
↓ 20 callersClassSpan
Represents a span (entity-mention) of text in a document, `Doc`.
src/refined/data_types/base_types.py:81
↓ 8 callersClassBatchElementToken
This class represents a token used to produce tensors for a forward pass of a nn. The `acc_sum` is used for mean pooling contiguous token for
src/refined/data_types/modelling_types.py:14
↓ 8 callersClassEntity
src/refined/data_types/base_types.py:63
↓ 8 callersClassLmdbImmutableDict
src/refined/resource_management/lmdb_wrapper.py:17
↓ 7 callersClassResourceManager
src/refined/resource_management/resource_manager.py:16
↓ 7 callersClassS3Manager
src/refined/resource_management/aws.py:17
↓ 6 callersClassBatchElement
Represents part of a document (`Doc`) that fits in single pass of a transformer model (< max_seq_len) as a singleton batch. If the document i
src/refined/data_types/modelling_types.py:110
↓ 3 callersClassBatchedElementsTns
Represents the tensors for a list/batch of `BatchElement` that have been collated (vertically stacking in dim=1). Equivalent to `BatchElement
src/refined/data_types/modelling_types.py:58
↓ 3 callersClassClassHandler
src/refined/doc_preprocessing/class_handler.py:8
↓ 3 callersClassInfix
Infix operators. The calling sequence for the infix is: x |op| y
src/refined/offline_data_generation/wiki_extractor.py:1728
↓ 3 callersClassMetrics
src/refined/evaluation/metrics.py:8
↓ 3 callersClassSimpleNamespace
src/refined/offline_data_generation/wiki_extractor.py:83
↓ 2 callersClassBatchElementTns
Batch element as a tuple of tensors. This tuple is for a single `BatchElement`.
src/refined/data_types/modelling_types.py:34
↓ 2 callersClassDataParallelReFinED
src/refined/torch_overrides/data_parallel_refined.py:47
↓ 2 callersClassExtractor
An extraction task on a article.
src/refined/offline_data_generation/wiki_extractor.py:538
↓ 2 callersClassFineTuningArgs
src/refined/training/fine_tune/fine_tune_args.py:13
↓ 2 callersClassFrame
src/refined/offline_data_generation/wiki_extractor.py:508
↓ 2 callersClassModelReturn
Result of RefinedModel(nn.Module).
src/refined/data_types/modelling_types.py:94
↓ 2 callersClassTemplateText
Fixed text of template
src/refined/offline_data_generation/wiki_extractor.py:447
↓ 2 callersClassWikidataMapper
src/refined/doc_preprocessing/wikidata_mapper.py:8
↓ 2 callersClassWikipediaDataset
This is a dataset reader for Wikipedia links dataset. It is a torch iterable dataset because it is too large to hold in memory. The datas
src/refined/dataset_reading/entity_linking/wikipedia_dataset.py:21
↓ 1 callersClassAdditionalEntity
src/refined/offline_data_generation/dataclasses_for_preprocessing.py:6
↓ 1 callersClassCandidateGeneratorExactMatch
src/refined/doc_preprocessing/candidate_generator.py:30
↓ 1 callersClassDatasets
src/refined/dataset_reading/entity_linking/dataset_factory.py:11
↓ 1 callersClassDateHandler
Handles resolving the string of a date mention to an instance of the Date class
src/refined/utilities/numeric_handling/date_utils.py:97
↓ 1 callersClassDescriptionEncoder
src/refined/model_components/description_encoder.py:12
↓ 1 callersClassDocDataset
Standard (go-to) dataset.
src/refined/dataset_reading/entity_linking/document_dataset.py:15
↓ 1 callersClassDownloadProgressBar
src/refined/offline_data_generation/preprocessing_utils.py:19
↓ 1 callersClassEDLayer
src/refined/model_components/ed_layer_2.py:11
↓ 1 callersClassEntityDisambiguation
src/refined/model_components/entity_disambiguation_layer.py:9
↓ 1 callersClassEntityTyping
src/refined/model_components/entity_typing_layer.py:7
↓ 1 callersClassLookupsInferenceOnly
src/refined/resource_management/data_lookups.py:17
↓ 1 callersClassMagicWords
One copy in each Extractor. @see https://doc.wikimedia.org/mediawiki-core/master/php/MagicWord_8php_source.html
src/refined/offline_data_generation/wiki_extractor.py:1541
↓ 1 callersClassMentionDetection
src/refined/model_components/mention_detection_layer.py:13
↓ 1 callersClassModelConfig
src/refined/model_components/config.py:29
↓ 1 callersClassNextFile
Synchronous generation of next available file name.
src/refined/offline_data_generation/wiki_extractor.py:2658
↓ 1 callersClassOutputSplitter
File-like object, that splits output to multiple files of a given max size.
src/refined/offline_data_generation/wiki_extractor.py:2690
↓ 1 callersClassPreprocessorInferenceOnly
Implementation of Preprocessor that uses memmaped objects and only contains data needed for inference. See `Preprocessor` for additional docu
src/refined/doc_preprocessing/preprocessor.py:114
↓ 1 callersClassRefined
This class provides methods to process text. It wraps around document preprocessor and the ReFinED model class.
src/refined/inference/processor.py:30
↓ 1 callersClassRefinedModel
Model for MD, ET, ED built on top of a transformer.
src/refined/model_components/refined_model.py:27
↓ 1 callersClassSpanDateHandler
For Span instances which have been identified by the MD layer as having type "DATE", resolve the dates from their text format (e.g. "12th Jun
src/refined/utilities/numeric_handling/date_utils.py:288
↓ 1 callersClassTemplate
A Template is a list of TemplateText or TemplateArgs
src/refined/offline_data_generation/wiki_extractor.py:398
↓ 1 callersClassTemplateArg
parameter to a template. Has a name and a default value, both of which are Templates.
src/refined/offline_data_generation/wiki_extractor.py:455
↓ 1 callersClassToken
This class represents a token in tokenized text.
src/refined/data_types/base_types.py:7
↓ 1 callersClassTrainingArgs
src/refined/training/train/training_args.py:17
ClassCandidateGenerator
src/refined/doc_preprocessing/candidate_generator.py:10
ClassCoNLLNER
src/refined/dataset_reading/mention_detection/conll_reader.py:30
ClassDate
src/refined/data_types/base_types.py:19
ClassDoc
Represents a document of text and spans.
src/refined/data_types/doc_types.py:15
ClassDocIterDataset
Use this class instead of `DocDataset` when the dataset is very large and does not fit in main memory.
src/refined/dataset_reading/entity_linking/document_dataset.py:94
ClassMentionDetector
src/refined/inference/standalone_md.py:29
ClassOntoNotesNER
src/refined/dataset_reading/mention_detection/ontonotes_reader.py:15
ClassPackage
Package Code and Dependencies into wheelhouse
package.py:13
ClassPreprocessor
Abstract class representing a preprocessor that provides methods to use resources such as Wikidata classes, and P(e|m) alias table. Implement
src/refined/doc_preprocessing/preprocessor.py:22
ClassReFinEDConfig
src/refined/data_types/modelling_types.py:248
ClassToken
src/refined/inference/standalone_md.py:18
ClassWebQSPNER
src/refined/dataset_reading/mention_detection/webqsp_reader.py:16