Skip to main content
Ctrl+K
Logo image

Site Navigation

  • Quick Start
  • API Reference

Site Navigation

  • Quick Start
  • API Reference

Section Navigation

  • nlptest.augmentation
    • nlptest.augmentation.AugmentRobustness
    • nlptest.augmentation.BaseAugmentaion
  • nlptest.datahandler
    • nlptest.datahandler.datasource
      • nlptest.datahandler.datasource.CSVDataset
      • nlptest.datahandler.datasource.ConllDataset
      • nlptest.datahandler.datasource.DataFactory
      • nlptest.datahandler.datasource.JSONDataset
      • nlptest.datahandler.datasource.JSONLDataset
    • nlptest.datahandler.format
      • nlptest.datahandler.format.BaseFormatter
      • nlptest.datahandler.format.Formatter
      • nlptest.datahandler.format.NEROutputFormatter
      • nlptest.datahandler.format.SequenceClassificationOutputFormatter
  • nlptest.modelhandler
    • nlptest.modelhandler.jsl_modelhandler
      • nlptest.modelhandler.jsl_modelhandler.PretrainedModelForNER
      • nlptest.modelhandler.jsl_modelhandler.PretrainedModelForTextClassification
    • nlptest.modelhandler.llm_modelhandler
      • nlptest.modelhandler.llm_modelhandler.PretrainedModelForQA
    • nlptest.modelhandler.modelhandler
      • nlptest.modelhandler.modelhandler.ModelFactory
    • nlptest.modelhandler.spacy_modelhandler
      • nlptest.modelhandler.spacy_modelhandler.PretrainedModelForNER
      • nlptest.modelhandler.spacy_modelhandler.PretrainedModelForTextClassification
    • nlptest.modelhandler.transformers_modelhandler
      • nlptest.modelhandler.transformers_modelhandler.PretrainedModelForNER
      • nlptest.modelhandler.transformers_modelhandler.PretrainedModelForTextClassification
  • nlptest.nlptest
    • nlptest.nlptest.Harness
  • nlptest.testrunner
    • nlptest.testrunner.BaseRunner
    • nlptest.testrunner.TestRunner
  • nlptest.transform
    • nlptest.transform.AccuracyTestFactory
    • nlptest.transform.BiasTestFactory
    • nlptest.transform.FairnessTestFactory
    • nlptest.transform.ITests
    • nlptest.transform.RepresentationTestFactory
    • nlptest.transform.RobustnessTestFactory
    • nlptest.transform.TestFactory
    • nlptest.transform.accuracy
      • nlptest.transform.accuracy.BaseAccuracy
      • nlptest.transform.accuracy.MinF1Score
      • nlptest.transform.accuracy.MinMacroF1Score
      • nlptest.transform.accuracy.MinMicroF1Score
      • nlptest.transform.accuracy.MinPrecisionScore
      • nlptest.transform.accuracy.MinRecallScore
      • nlptest.transform.accuracy.MinWeightedF1Score
    • nlptest.transform.bias
      • nlptest.transform.bias.BaseBias
      • nlptest.transform.bias.CountryEconomicBias
      • nlptest.transform.bias.EthnicityNameBias
      • nlptest.transform.bias.GenderPronounBias
      • nlptest.transform.bias.ReligionBias
    • nlptest.transform.fairness
      • nlptest.transform.fairness.get_gendered_data
      • nlptest.transform.fairness.BaseFairness
      • nlptest.transform.fairness.MaxGenderF1Score
      • nlptest.transform.fairness.MinGenderF1Score
    • nlptest.transform.representation
      • nlptest.transform.representation.BaseRepresentation
      • nlptest.transform.representation.CountryEconomicRepresentation
      • nlptest.transform.representation.EthnicityRepresentation
      • nlptest.transform.representation.GenderRepresentation
      • nlptest.transform.representation.LabelRepresentation
      • nlptest.transform.representation.ReligionRepresentation
    • nlptest.transform.robustness
      • nlptest.transform.robustness.AddContext
      • nlptest.transform.robustness.AddContraction
      • nlptest.transform.robustness.AddPunctuation
      • nlptest.transform.robustness.AddTypo
      • nlptest.transform.robustness.BaseRobustness
      • nlptest.transform.robustness.ConvertAccent
      • nlptest.transform.robustness.LowerCase
      • nlptest.transform.robustness.StripPunctuation
      • nlptest.transform.robustness.SwapEntities
      • nlptest.transform.robustness.TitleCase
      • nlptest.transform.robustness.UpperCase
    • nlptest.transform.utils
      • nlptest.transform.utils.check_name
      • nlptest.transform.utils.create_terminology
      • nlptest.transform.utils.get_country_economic_representation_dict
      • nlptest.transform.utils.get_entity_representation_proportions
      • nlptest.transform.utils.get_ethnicity_representation_dict
      • nlptest.transform.utils.get_label_representation_dict
      • nlptest.transform.utils.get_religion_name_representation_dict
      • nlptest.transform.utils.get_substitution_names
  • nlptest.utils
    • nlptest.utils.custom_types
      • nlptest.utils.custom_types.helpers
        • nlptest.utils.custom_types.helpers.Span
        • nlptest.utils.custom_types.helpers.Transformation
      • nlptest.utils.custom_types.output
        • nlptest.utils.custom_types.output.MaxScoreOutput
        • nlptest.utils.custom_types.output.MinScoreOutput
        • nlptest.utils.custom_types.output.NEROutput
        • nlptest.utils.custom_types.output.SequenceClassificationOutput
      • nlptest.utils.custom_types.predictions
        • nlptest.utils.custom_types.predictions.NERPrediction
        • nlptest.utils.custom_types.predictions.SequenceLabel
      • nlptest.utils.custom_types.sample
        • nlptest.utils.custom_types.sample.BaseQASample
        • nlptest.utils.custom_types.sample.BaseSample
        • nlptest.utils.custom_types.sample.MaxScoreSample
        • nlptest.utils.custom_types.sample.MinScoreSample
        • nlptest.utils.custom_types.sample.NERSample
        • nlptest.utils.custom_types.sample.QASample
        • nlptest.utils.custom_types.sample.SequenceClassificationSample
    • nlptest.utils.gender_classifier
      • nlptest.utils.gender_classifier.GenderClassifier
    • nlptest.utils.lib_manager
      • nlptest.utils.lib_manager.try_import_lib
  • nlptest
  • nlptest.modelhandler
  • nlptest.modelhandler.transformers_modelhandler

nlptest.modelhandler.transformers_modelhandler#

Classes

PretrainedModelForNER(model)

param model:

Pretrained HuggingFace NER pipeline for predictions.

PretrainedModelForTextClassification(model)

model#

previous

nlptest.modelhandler.spacy_modelhandler.PretrainedModelForTextClassification

next

nlptest.modelhandler.transformers_modelhandler.PretrainedModelForNER
On this page
  • model

© Copyright 2023, John Snow Labs.

Created using Sphinx 6.1.3.

Built with the PyData Sphinx Theme 0.13.3.