.coveragerc
.gitignore
.readthedocs.yaml
CODE.md
CONTRIBUTING.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
pytest.ini
setup.cfg
setup.py
tox.ini
.github/workflows/build_test_publish.yml
docs/Makefile
docs/conf.py
docs/index.rst
docs/make.bat
docs/requirements.txt
docs/img/bird_trial_display.gif
docs/img/birds-16_emb_3d.gif
docs/img/birds-16_thumbs.png
docs/img/favicon.ico
docs/img/full_logo_300.png
docs/img/model_2c1_example_v2.png
docs/img/model_flow_1_v2.png
docs/img/n-choose-k_label_convention.png
docs/src/refs.bib
docs/src/beginner_tutorial/beginner_tutorial.ipynb
docs/src/design/mask_zero.rst
docs/src/design/model_restarts.rst
docs/src/features/data.rst
docs/src/features/faq.rst
docs/src/features/hosted.rst
docs/src/features/models.rst
docs/src/features/overview.rst
docs/src/features/your_data.rst
docs/src/first_steps/intro.rst
docs/src/first_steps/minimal_example.py
docs/src/first_steps/quick_start.rst
docs/src/tutorials/active_learning.rst
docs/src/tutorials/vi.rst
examples/README.md
examples/hosted_datasets.py
examples/rank/mle_1g.py
examples/rank/mle_3g.py
examples/rank/understanding_information_gain.py
examples/rank/vi_1g.py
examples/rank/vi_1g_nonneg.py
examples/rank/vi_3g.py
examples/rank/vi_3ge.py
examples/rank/vi_birds_16.py
examples/rate/mle_1g.py
src/psiz/__init__.py
src/psiz/version.py
src/psiz.egg-info/PKG-INFO
src/psiz.egg-info/SOURCES.txt
src/psiz.egg-info/dependency_links.txt
src/psiz.egg-info/requires.txt
src/psiz.egg-info/top_level.txt
src/psiz/agents/__init__.py
src/psiz/agents/base.py
src/psiz/agents/rank_agent.py
src/psiz/catalog/__init__.py
src/psiz/catalog/catalog.py
src/psiz/catalog/load_catalog.py
src/psiz/datasets/__init__.py
src/psiz/datasets/load_dataset.py
src/psiz/keras/__init__.py
src/psiz/keras/restarter.py
src/psiz/keras/sparse_dispatcher.py
src/psiz/keras/callbacks/__init__.py
src/psiz/keras/callbacks/early_stopping_re.py
src/psiz/keras/callbacks/tensor_board_re.py
src/psiz/keras/constraints/__init__.py
src/psiz/keras/constraints/center.py
src/psiz/keras/constraints/greater_equal_than.py
src/psiz/keras/constraints/greater_than.py
src/psiz/keras/constraints/less_equal_than.py
src/psiz/keras/constraints/less_than.py
src/psiz/keras/constraints/min_max.py
src/psiz/keras/constraints/non_neg_norm.py
src/psiz/keras/initializers/__init__.py
src/psiz/keras/initializers/dirichlet.py
src/psiz/keras/initializers/random_scale_mvn.py
src/psiz/keras/initializers/softplus_uniform.py
src/psiz/keras/layers/__init__.py
src/psiz/keras/layers/gate.py
src/psiz/keras/layers/gate_multi.py
src/psiz/keras/layers/variational.py
src/psiz/keras/layers/behaviors/__init__.py
src/psiz/keras/layers/behaviors/base.py
src/psiz/keras/layers/behaviors/rank.py
src/psiz/keras/layers/behaviors/rate.py
src/psiz/keras/layers/behaviors/sort.py
src/psiz/keras/layers/distances/__init__.py
src/psiz/keras/layers/distances/mink.py
src/psiz/keras/layers/distances/mink_stochastic.py
src/psiz/keras/layers/distances/mink_variational.py
src/psiz/keras/layers/embeddings/__init__.py
src/psiz/keras/layers/embeddings/gamma_diag.py
src/psiz/keras/layers/embeddings/laplace_diag.py
src/psiz/keras/layers/embeddings/loc_scale.py
src/psiz/keras/layers/embeddings/log_normal_diag.py
src/psiz/keras/layers/embeddings/logit_normal_diag.py
src/psiz/keras/layers/embeddings/normal_diag.py
src/psiz/keras/layers/embeddings/shared.py
src/psiz/keras/layers/embeddings/stochastic_embedding.py
src/psiz/keras/layers/embeddings/trunc_normal_diag.py
src/psiz/keras/layers/embeddings/variational.py
src/psiz/keras/layers/kernels/__init__.py
src/psiz/keras/layers/kernels/distance_based.py
src/psiz/keras/layers/similarities/__init__.py
src/psiz/keras/layers/similarities/exponential.py
src/psiz/keras/layers/similarities/heavy_tailed.py
src/psiz/keras/layers/similarities/inverse.py
src/psiz/keras/layers/similarities/students_t.py
src/psiz/keras/models/__init__.py
src/psiz/keras/models/psych_embedding.py
src/psiz/keras/models/rank.py
src/psiz/keras/models/rate.py
src/psiz/keras/models/sort.py
src/psiz/keras/regularizers/__init__.py
src/psiz/keras/regularizers/attention_entropy.py
src/psiz/keras/regularizers/squeeze.py
src/psiz/keras/regularizers/stimulus_normed_l1.py
src/psiz/mplot/__init__.py
src/psiz/mplot/bvn_ellipse.py
src/psiz/mplot/embedding_input_dimension.py
src/psiz/mplot/embedding_output_dimension.py
src/psiz/mplot/hdi_bvn.py
src/psiz/mplot/heatmap_embeddings.py
src/psiz/tf/ops/__init__.py
src/psiz/tf/ops/wpnorm.py
src/psiz/tfp/__init__.py
src/psiz/tfp/distributions/__init__.py
src/psiz/tfp/distributions/gamma.py
src/psiz/tfp/distributions/inv_softplus_normal.py
src/psiz/tfp/distributions/truncated_normal.py
src/psiz/trials/__init__.py
src/psiz/trials/load_trials.py
src/psiz/trials/sample_qr_sets.py
src/psiz/trials/stack.py
src/psiz/trials/experimental/__init__.py
src/psiz/trials/experimental/trial_component.py
src/psiz/trials/experimental/trial_dataset.py
src/psiz/trials/experimental/unravel_timestep.py
src/psiz/trials/experimental/contents/__init__.py
src/psiz/trials/experimental/contents/content.py
src/psiz/trials/experimental/contents/rank_similarity.py
src/psiz/trials/experimental/contents/rate_similarity.py
src/psiz/trials/experimental/outcomes/__init__.py
src/psiz/trials/experimental/outcomes/continuous.py
src/psiz/trials/experimental/outcomes/outcome.py
src/psiz/trials/experimental/outcomes/sparse_categorical.py
src/psiz/trials/information_gain/__init__.py
src/psiz/trials/information_gain/ig_categorical.py
src/psiz/trials/information_gain/ig_model_categorical.py
src/psiz/trials/similarity/__init__.py
src/psiz/trials/similarity/docket_generator.py
src/psiz/trials/similarity/similarity_trials.py
src/psiz/trials/similarity/rank/__init__.py
src/psiz/trials/similarity/rank/active_rank.py
src/psiz/trials/similarity/rank/random_rank.py
src/psiz/trials/similarity/rank/rank_docket.py
src/psiz/trials/similarity/rank/rank_observations.py
src/psiz/trials/similarity/rank/rank_trials.py
src/psiz/trials/similarity/rate/__init__.py
src/psiz/trials/similarity/rate/random_rate.py
src/psiz/trials/similarity/rate/rate_docket.py
src/psiz/trials/similarity/rate/rate_observations.py
src/psiz/trials/similarity/rate/rate_trials.py
src/psiz/utils/__init__.py
src/psiz/utils/affine_mvn.py
src/psiz/utils/choice_wo_replace.py
src/psiz/utils/expand_dim_repeat.py
src/psiz/utils/fit_tracker.py
src/psiz/utils/generate_group_matrix.py
src/psiz/utils/pairwise_index_dataset.py
src/psiz/utils/pairwise_similarity.py
src/psiz/utils/procrustes.py
src/psiz/utils/progress_bar_re.py
src/psiz/utils/random_combinations.py
src/psiz/utils/rotation_matrix.py
src/psiz/utils/standard_split.py
src/psiz/utils/stratified_group_kfold.py
tests/conftest.py
tests/test_catalog.py
tests/test_datasets.py
tests/agents/test_rank_agent.py
tests/examples/test_mle_1g.py
tests/examples/test_vi_1g.py
tests/keras/test_sparse_dispatcher.py
tests/keras/constraints/test_center.py
tests/keras/constraints/test_greater_than.py
tests/keras/constraints/test_less_equal_than.py
tests/keras/constraints/test_less_than.py
tests/keras/constraints/test_min_max.py
tests/keras/constraints/test_non_neg_norm.py
tests/keras/initializers/test_dirichlet.py
tests/keras/layers/conftest.py
tests/keras/layers/test_gate.py
tests/keras/layers/test_gate_multi.py
tests/keras/layers/distances/test_mink_deterministic.py
tests/keras/layers/distances/test_mink_stochastic.py
tests/keras/layers/distances/test_mink_variational.py
tests/keras/layers/embeddings/conftest.py
tests/keras/layers/embeddings/test_embedding_variational.py
tests/keras/layers/embeddings/test_stochastic_embeddings.py
tests/keras/layers/kernels/test_distance_based.py
tests/keras/layers/similarities/test_exponential.py
tests/keras/layers/similarities/test_heavy_tailed.py
tests/keras/layers/similarities/test_inverse.py
tests/keras/layers/similarities/test_students_t.py
tests/keras/models/conftest.py
tests/keras/models/test_rank.py
tests/keras/models/test_rate.py
tests/mplot/test_bvn_ellipse.py
tests/mplot/test_embedding_input_dimension.py
tests/mplot/test_embedding_output_dimension.py
tests/mplot/test_heatmap_embeddings.py
tests/tfp/test_truncated_normal.py
tests/trials/conftest.py
tests/trials/test_sample_qr_sets.py
tests/trials/test_trial_dataset.py
tests/trials/contents/test_rank_similarity.py
tests/trials/contents/test_rate_similarity.py
tests/trials/information_gain/test_ig_categorical.py
tests/trials/information_gain/test_ig_model_categorical.py
tests/trials/outcomes/test_continuous.py
tests/trials/outcomes/test_sparse_categorical.py
tests/trials/similarity/test_random_rank.py
tests/trials/similarity/test_rank_trials.py
tests/trials/similarity/test_rate_trials.py
tests/trials/similarity/test_similarity_trials.py
tests/utils/test_affine_mvn.py
tests/utils/test_choice_wo_replace.py
tests/utils/test_expand_dim_repeat.py
tests/utils/test_generate_group_matrix.py
tests/utils/test_pairwise_index_dataset.py
tests/utils/test_pairwise_similarity.py
tests/utils/test_procrustes.py
tests/utils/test_random_combinations.py
tests/utils/test_strat_group_kfold.py