autointent.modules#

Heart of the library with different intent classification methods implemented.

Submodules#

Attributes#

Classes#

BaseDecision

Base class for decision modules.

BaseEmbedding

Base class for embedding modules.

BaseModule

Base module for all intent classification modules.

BaseRegex

Base class for rule-based modules.

BaseScorer

Abstract base class for scoring modules.

AdaptiveDecision

Decision for multi-label classification using adaptive thresholds.

ArgmaxDecision

Argmax decision module.

JinoosDecision

Jinoos predictor module.

ThresholdDecision

Threshold predictor module.

TunableDecision

Tunable predictor module.

LogregAimedEmbedding

Module for configuring embeddings optimized for linear classification.

RetrievalAimedEmbedding

Module for configuring embeddings optimized for retrieval tasks.

SimpleRegex

Regular expressions based intent detection module.

BERTLoRAScorer

BERTLoRAScorer class for transformer-based classification with LoRA (Low-Rank Adaptation).

BertScorer

Scoring module for transformer-based classification using BERT models.

BiEncoderDescriptionScorer

Bi-encoder description scorer for zero-shot intent classification.

CatBoostScorer

CatBoost scorer using either external embeddings or CatBoost's own BoW encoding.

CNNScorer

Convolutional Neural Network (CNN) scorer for intent classification.

CrossEncoderDescriptionScorer

Cross-encoder description scorer for zero-shot intent classification.

DNNCScorer

Scoring module for intent classification using discriminative nearest neighbor classification.

KNNScorer

K-nearest neighbors (KNN) scorer for intent classification.

LinearScorer

Scoring module for linear classification using logistic regression.

LLMDescriptionScorer

LLM-based description scorer for zero-shot intent classification using structured output.

MLKnnScorer

Multi-label k-nearest neighbors (ML-KNN) scorer.

PTuningScorer

PEFT P-tuning scorer.

RerankScorer

Re-ranking scorer using a cross-encoder for intent classification.

RNNScorer

Recurrent Neural Network (RNN) scorer for intent classification.

SklearnScorer

Scoring module for classification using sklearn classifiers.

Package Contents#

autointent.modules.T#
autointent.modules.REGEX_MODULES: dict[str, type[base.BaseRegex]]#
autointent.modules.EMBEDDING_MODULES: dict[str, type[base.BaseEmbedding]]#
autointent.modules.SCORING_MODULES: dict[str, type[base.BaseScorer]]#
autointent.modules.DECISION_MODULES: dict[str, type[base.BaseDecision]]#