autointent.schemas.node_validation#

Schemes.

Attributes#

Classes#

ParamSpace

Base class for search space used in optuna.

ParamSpaceInt

Integer parameter search space configuration.

ParamSpaceFloat

Float parameter search space configuration.

DecisionNodeValidator

Search space configuration for the Decision node.

EmbeddingNodeValidator

Search space configuration for the Embedding node.

ScoringNodeValidator

Search space configuration for the Scoring node.

RegexNodeValidator

Search space configuration for the Regexp node.

SearchSpaceConfig

Search space configuration.

OptimizationSearchSpaceConfig

Optimizer configuration.

Functions#

unwrap_annotated(tp)

Unwrap the Annotated type to get the actual type.

type_matches(target, tp)

Recursively check if the target type is present in the given type.

get_optuna_class(param_type)

Get the Optuna class for the given parameter type.

generate_models_and_union_type_for_classes(classes)

Dynamically generates Pydantic models for class constructors and creates a union type.

Module Contents#

autointent.schemas.node_validation.ParamSpaceT#
autointent.schemas.node_validation.DecisionMetrics#
type autointent.schemas.node_validation.EmbeddingMetrics = Literal[tuple(EmbeddingNodeInfo.metrics_available.keys())]#
type autointent.schemas.node_validation.ScoringMetrics = Literal[tuple(ScoringNodeInfo.metrics_available.keys())]#
type autointent.schemas.node_validation.RegexpMetrics = Literal[tuple(RegexNodeInfo.metrics_available.keys())]#
type autointent.schemas.node_validation.NodeValidatorType = EmbeddingNodeValidator | ScoringNodeValidator | DecisionNodeValidator | RegexNodeValidator#
type autointent.schemas.node_validation.SearchSpaceType = DecisionSearchSpaceType | EmbeddingSearchSpaceType | ScoringSearchSpaceType | RegexpSearchSpaceType#