def compile()

in src/lic/ppl/experimental/inference_compilation/ic_infer.py [0:0]

• 15 lines of code
• 1 McCabe index (conditional complexity)

    def compile(  # noqa: C901
        self,
        observation_keys: Sequence[RVIdentifier],
        num_worlds: int = 100,
        batch_size: int = 16,
        optimizer_func=lambda parameters: optim.Adam(parameters),
        node_id_embedding_dim: Optional[int] = None,
        node_embedding_dim: Optional[int] = None,
        obs_embedding_dim: Optional[int] = None,
        mb_embedding_dim: Optional[int] = None,
        mb_num_layers: Optional[int] = None,
        node_proposal_num_layers: Optional[int] = None,
        entropy_regularization_coefficient: Optional[float] = None,
        gmm_num_components: Optional[int] = None,
        query_hints: Optional[Sequence[RVIdentifier]] = None,