Criticism

pyroed.criticism.criticize(schema: Dict[str, List[Optional[str]]], feature_blocks: List[List[str]], experiment: Dict[str, torch.Tensor], test_data: Dict[str, torch.Tensor], *, feature_fn: Optional[Callable[[torch.Tensor], torch.Tensor]] = None, response_type: str = 'unit_interval', inference: str = 'svi', num_posterior_samples: int = 11, mcmc_num_samples: int = 500, mcmc_warmup_steps: int = 500, mcmc_num_chains: int = 1, svi_num_steps: int = 501, svi_reparam: bool = True, thompson_temperature: float = 1.0, jit_compile: bool = False, log_every: int = 100, filename: Optional[str] = None)

Plots observed versus predicted responses on a held out test set.

Parameters

• schema (OrderedDict) – A schema dict.

• feature_blocks (list) – A list of choice blocks for linear regression.

• experiment (dict) – A dict containing training data.

• test_data (dict) – A dict containing held out test data.