lale.search.schema2search_space module

class lale.search.schema2search_space.FreqsWrapper(base: Optional[Dict[str, Dict[str, int]]])[source]

Bases: object

base: Optional[Dict[str, Dict[str, int]]]
exception lale.search.schema2search_space.OperatorSchemaError(sub_path: Any, message: Optional[str] = None)[source]

Bases: VisitorPathError

get_message_str() → str[source]
class lale.search.schema2search_space.SearchSpaceOperatorVisitor(pgo: Optional[Dict[str, Dict[str, Dict[str, int]]]] = None, data_schema: Optional[Dict[str, Any]] = None)[source]

Bases: Visitor

JsonSchemaToSearchSpaceHelper(longName: str, path: str, schema: Dict[str, Any], relevantFields: Optional[Set[str]], pgo_freqs: Optional[Union[FreqsWrapper, Dict[str, int]]] = None, sub_space: bool = True) → Dict[str, SearchSpace][source]
data_schema: Optional[Dict[str, Any]]
pgo: Optional[Dict[str, Dict[str, Dict[str, int]]]]
classmethod run(op: PlannedOperator, pgo: Optional[Dict[str, Dict[str, Dict[str, int]]]] = None, data_schema: Optional[Dict[str, Any]] = None) → SearchSpace[source]
schemaToSearchSpace(longName: str, name: str, schema: Dict[str, Any]) → Optional[SearchSpace][source]
schemaToSearchSpaceHelper(longName, schema: Optional[Dict[str, Any]], relevantFields: Optional[Set[str]], pgo_freqs: Optional[Union[FreqsWrapper, Dict[str, int]]] = None, sub_space: bool = True) → Optional[SearchSpace][source]
schemaToSearchSpaceHelper_(longName, path: str, schema: Dict[str, Any], relevantFields: Optional[Set[str]], pgo_freqs: Optional[Union[FreqsWrapper, Dict[str, int]]] = None, sub_space: bool = True) → Optional[SearchSpace][source]
schemaToSimplifiedAndSearchSpace(longName: str, name: str, schema: Dict[str, Any]) → Tuple[Optional[Dict[str, Any]], Optional[SearchSpace]][source]
visitOperatorChoice(op: OperatorChoice) → SearchSpace[source]
visitPlannedIndividualOp(op: PlannedIndividualOp) → SearchSpace[source]
visitPlannedPipeline(op: PlannedPipeline) → SearchSpace[source]
visitTrainableIndividualOp(op: PlannedIndividualOp) → SearchSpace
visitTrainablePipeline(op: PlannedPipeline) → SearchSpace
visitTrainedIndividualOp(op: PlannedIndividualOp) → SearchSpace
visitTrainedPipeline(op: PlannedPipeline) → SearchSpace
lale.search.schema2search_space.add_sub_space(space, k, v)[source]

Given a search space and a “key”, if the defined subschema does not exist, set it to be the constant v space

lale.search.schema2search_space.asFreqs(part: Optional[Union[FreqsWrapper, Dict[str, int]]]) → Optional[Iterable[Tuple[Any, int]]][source]
lale.search.schema2search_space.freqs_wrapper_lookup(part: Optional[Union[FreqsWrapper, Dict[str, int]]], k: str) → Optional[Union[FreqsWrapper, Dict[str, int]]][source]
lale.search.schema2search_space.get_default(schema) → Optional[Any][source]
lale.search.schema2search_space.op_to_search_space(op: PlannedOperator, pgo: Optional[Dict[str, Dict[str, Dict[str, int]]]] = None, data_schema: Optional[Dict[str, Any]] = None) → SearchSpace[source]

Given an operator, this method compiles its schemas into a SearchSpace

lale.search.schema2search_space.pgo_lookup(pgo: Optional[Dict[str, Dict[str, Dict[str, int]]]], name: str) → Optional[FreqsWrapper][source]