autointent.modules#
Heart of the library with different intent classification methods implemented.
Submodules#
Attributes#
Classes#
Base class for decision modules. |
|
Base class for embedding modules. |
|
Base module for all intent classification modules. |
|
Base class for rule-based modules. |
|
Abstract base class for scoring modules. |
|
Decision for multi-label classification using adaptive thresholds. |
|
Argmax decision module. |
|
Jinoos predictor module. |
|
Threshold predictor module. |
|
Tunable predictor module. |
|
Module for configuring embeddings optimized for linear classification. |
|
Module for configuring embeddings optimized for retrieval tasks. |
|
Regular expressions based intent detection module. |
|
BERTLoRAScorer class for transformer-based classification with LoRA (Low-Rank Adaptation). |
|
Scoring module for transformer-based classification using BERT models. |
|
Bi-encoder description scorer for zero-shot intent classification. |
|
CatBoost scorer using either external embeddings or CatBoost's own BoW encoding. |
|
Convolutional Neural Network (CNN) scorer for intent classification. |
|
Cross-encoder description scorer for zero-shot intent classification. |
|
Scoring module for intent classification using discriminative nearest neighbor classification. |
|
K-nearest neighbors (KNN) scorer for intent classification. |
|
Scoring module for linear classification using logistic regression. |
|
LLM-based description scorer for zero-shot intent classification using structured output. |
|
Multi-label k-nearest neighbors (ML-KNN) scorer. |
|
PEFT P-tuning scorer. |
|
Re-ranking scorer using a cross-encoder for intent classification. |
|
Recurrent Neural Network (RNN) scorer for intent classification. |
|
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]]#