autointent.modules#

Heart of the library with different intent classification methods implemented.

Submodules#

Attributes#

Classes#

BaseModule

Base module for all intent classification 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.

GCNScorer

Graph Convolutional Network (GCN) scorer for intent 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]]#