djai.model.models.ml.hugging_face.zero_shot_classification module

DjAI Pre-Trained Hugging Face Zero-Shot Classifier Model class.

class djai.model.models.ml.hugging_face.zero_shot_classification.PreTrainedHuggingFaceZeroShotClassifier(*args, **kwargs)

Bases: djai.model.models.ml.hugging_face.base.PreTrainedHuggingFaceTransformer

DjAI Pre-Trained Hugging Face Zero-Shot Classifier Model class.

exception DoesNotExist

Bases: djai.model.models.ml.hugging_face.base.PreTrainedHuggingFaceTransformer.DoesNotExist

exception MultipleObjectsReturned

Bases: djai.model.models.ml.hugging_face.base.PreTrainedHuggingFaceTransformer.MultipleObjectsReturned

gradio_ui = Gradio Interface for: <lambda> ------------------------------ inputs: |-Textbox(label="Text to Classify") |-Dataframe(label="Candidate Labels") |-Textbox(label="Hypothesis Format") |-Checkbox(label="Multi-Label?") outputs: |-Label(label="Label Probabilities")

polymorphic_primary_key_name = 'uuid'

polymorphic_super_sub_accessors_replaced = False

predict(text_or_texts: Union[str, collections.abc.Sequence[str]], candidate_labels: List[str], hypothesis_template: str = 'This example is {}.', multi_label: bool = False) → Union[Dict[str, float], List[Dict[str, float]]]

Zero-Shot Classification of Text(s).

pretrainedhuggingfacetransformer_ptr

Accessor to the related object on the forward side of a one-to-one relation.

In the example:

class Restaurant(Model):
    place = OneToOneField(Place, related_name='restaurant')

Restaurant.place is a ForwardOneToOneDescriptor instance.

pretrainedhuggingfacetransformer_ptr_id