Skip to main content
Back to top
Ctrl
+
K
aimz
User Guide
Examples
FAQs
API Reference
More
Development
Changelog
Search
Ctrl
+
K
System Settings
Light
Dark
GitHub
Search
Ctrl
+
K
User Guide
Examples
FAQs
API Reference
Development
Changelog
System Settings
Light
Dark
GitHub
Collapse Sidebar
Expand Sidebar
Section Navigation
KernelSpec
ImpactModel
aimz.ImpactModel
aimz.ImpactModel.inference
aimz.ImpactModel.kernel
aimz.ImpactModel.kernel_spec
aimz.ImpactModel.param_input
aimz.ImpactModel.param_output
aimz.ImpactModel.posterior
aimz.ImpactModel.rng_key
aimz.ImpactModel.temp_dir
aimz.ImpactModel.vi_result
aimz.ImpactModel.train_on_batch
aimz.ImpactModel.fit_on_batch
aimz.ImpactModel.fit
aimz.ImpactModel.set_posterior_sample
aimz.ImpactModel.is_fitted
aimz.ImpactModel.predict_on_batch
aimz.ImpactModel.predict
aimz.ImpactModel.log_likelihood
aimz.ImpactModel.sample_prior_predictive_on_batch
aimz.ImpactModel.sample_prior_predictive
aimz.ImpactModel.sample
aimz.ImpactModel.sample_posterior_predictive_on_batch
aimz.ImpactModel.sample_posterior_predictive
aimz.ImpactModel.estimate_effect
aimz.ImpactModel.cleanup
aimz.ImpactModel.cleanup_models
ArrayDataset
aimz.utils.data.ArrayDataset
ArrayLoader
aimz.utils.data.ArrayLoader
aimz.utils.data.ArrayLoader.pad_array
MLflow Integration
aimz.mlflow.autolog
aimz.mlflow.get_default_conda_env
aimz.mlflow.get_default_pip_requirements
aimz.mlflow.load_model
aimz.mlflow.log_model
aimz.mlflow.save_model
API Reference
ArrayDataset
ArrayDataset
#
ArrayDataset
(*[, to_jax])
Dataset for named arrays.
Edit on GitHub