autointent.configs

Dataclasses for the configuration of the autointent.Embedder and other objects.

Attributes

EmbedderConfig

Classes

HashingVectorizerEmbeddingConfig	Configuration for HashingVectorizer based embeddings from sklearn.
OpenaiEmbeddingConfig	Configuration for OpenAI based embeddings.
SentenceTransformerEmbeddingConfig	Configuration for Sentence Transformer based embeddings.
TaskTypeEnum	Enum for different types of prompts.
VllmEmbeddingConfig	Configuration for vLLM-based embeddings.
InferenceNodeConfig	Configuration for the inference node.
DataConfig	Configuration for the data used in the optimization process.
HPOConfig	Configuration for hyperparameter optimization using Optuna.
LoggingConfig	Configuration for the logging.
TorchTrainingConfig	!!! abstract "Usage Documentation"
VocabConfig	!!! abstract "Usage Documentation"
CrossEncoderConfig	!!! abstract "Usage Documentation"
EarlyStoppingConfig	!!! abstract "Usage Documentation"
EmbedderFineTuningConfig	!!! abstract "Usage Documentation"
HFModelConfig	!!! abstract "Usage Documentation"
TokenizerConfig	!!! abstract "Usage Documentation"
FaissConfig	!!! abstract "Usage Documentation"
OpenSearchConfig	!!! abstract "Usage Documentation"
VectorIndexConfig	!!! abstract "Usage Documentation"

Functions

get_default_embedder_config(**kwargs)
initialize_embedder_config(values)
get_default_hfmodel_config()
get_default_vector_index_config()

Package Contents

type autointent.configs.EmbedderConfig = SentenceTransformerEmbeddingConfig | OpenaiEmbeddingConfig | HashingVectorizerEmbeddingConfig | VllmEmbeddingConfig | BaseEmbedderConfig