.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_2rank1_example_v5.png
docs/img/model_flow_1task_v4.png
docs/img/n-choose-k_label_convention.png
docs/src/bibliography.rst
docs/src/refs.bib
docs/src/api_reference/psiz_data.rst
docs/src/api_reference/psiz_keras.rst
docs/src/api_reference/psiz_mplot.rst
docs/src/api_reference/psiz_tf.rst
docs/src/api_reference/psiz_utils.rst
docs/src/design/mask_zero.rst
docs/src/design/model_restarts.rst
docs/src/ecosystem/datasets.rst
docs/src/features/data.rst
docs/src/features/faq.rst
docs/src/first_steps/getting_started.rst
docs/src/tutorials/active_learning.rst
docs/src/tutorials/beginner_tutorial_part1.ipynb
docs/src/tutorials/beginner_tutorial_part2.ipynb
docs/src/tutorials/tuning.rst
docs/src/tutorials/vi.rst
examples/README.md
examples/rank/mle_1g.py
examples/rank/mle_3g.py
examples/rank/understanding_information_gain.py
examples/rank/vi_1g.py
examples/rank/vi_3ge.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/data/__init__.py
src/psiz/data/dataset.py
src/psiz/data/dataset_component.py
src/psiz/data/sample_qr_sets.py
src/psiz/data/unravel_timestep.py
src/psiz/data/contents/__init__.py
src/psiz/data/contents/categorize.py
src/psiz/data/contents/content.py
src/psiz/data/contents/rank.py
src/psiz/data/contents/rate.py
src/psiz/data/generators/__init__.py
src/psiz/data/generators/content_generator.py
src/psiz/data/generators/rank_generator.py
src/psiz/data/generators/experimental/weighted_rank.py
src/psiz/data/groups/__init__.py
src/psiz/data/groups/group.py
src/psiz/data/outcomes/__init__.py
src/psiz/data/outcomes/continuous.py
src/psiz/data/outcomes/outcome.py
src/psiz/data/outcomes/sparse_categorical.py
src/psiz/keras/__init__.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/softplus_uniform.py
src/psiz/keras/layers/__init__.py
src/psiz/keras/layers/combiner.py
src/psiz/keras/layers/drop.py
src/psiz/keras/layers/splitter.py
src/psiz/keras/layers/variational.py
src/psiz/keras/layers/activations/__init__.py
src/psiz/keras/layers/activations/exponential.py
src/psiz/keras/layers/activations/heavy_tailed.py
src/psiz/keras/layers/activations/inverse.py
src/psiz/keras/layers/activations/students_t.py
src/psiz/keras/layers/behaviors/__init__.py
src/psiz/keras/layers/behaviors/logistic.py
src/psiz/keras/layers/behaviors/soft_rank.py
src/psiz/keras/layers/behaviors/soft_rank_base.py
src/psiz/keras/layers/behaviors/experimental/__init__.py
src/psiz/keras/layers/behaviors/experimental/alcove_cell.py
src/psiz/keras/layers/behaviors/experimental/soft_rank_cell.py
src/psiz/keras/layers/embeddings/__init__.py
src/psiz/keras/layers/embeddings/embedding_shared.py
src/psiz/keras/layers/embeddings/embedding_variational.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/stochastic_embedding.py
src/psiz/keras/layers/embeddings/trunc_normal_diag.py
src/psiz/keras/layers/gates/braid_gate.py
src/psiz/keras/layers/gates/gate.py
src/psiz/keras/layers/gates/gate_adapter.py
src/psiz/keras/layers/proximities/__init__.py
src/psiz/keras/layers/proximities/minkowski.py
src/psiz/keras/layers/proximities/minkowski_stochastic.py
src/psiz/keras/layers/proximities/minkowski_variational.py
src/psiz/keras/layers/proximities/proximity.py
src/psiz/keras/layers/proximities/experimental/cosine_similarity.py
src/psiz/keras/layers/proximities/experimental/generalized_inner_product.py
src/psiz/keras/layers/proximities/experimental/inner_product.py
src/psiz/keras/models/__init__.py
src/psiz/keras/models/stochastic_model.py
src/psiz/keras/ops/__init__.py
src/psiz/keras/ops/expand_dim_repeat.py
src/psiz/keras/ops/ig_categorical.py
src/psiz/keras/ops/ig_model_categorical.py
src/psiz/keras/ops/wpnorm.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/tfp/__init__.py
src/psiz/tfp/unpack_mvn.py
src/psiz/tfp/distributions/__init__.py
src/psiz/tfp/distributions/inv_softplus_normal.py
src/psiz/utils/__init__.py
src/psiz/utils/affine_mvn.py
src/psiz/utils/choice_wo_replace.py
src/psiz/utils/m_prefer_n.py
src/psiz/utils/pairwise_indices.py
src/psiz/utils/procrustes_rotation.py
src/psiz/utils/random_combinations.py
src/psiz/utils/rotation_matrix.py
src/psiz/utils/sort_based_mask.py
src/psiz/utils/standard_split.py
src/psiz/utils/stratified_group_kfold.py
tests/examples/test_rank_mle_1g.py
tests/psiz/data/conftest.py
tests/psiz/data/test_dataset.py
tests/psiz/data/test_sample_qr_sets.py
tests/psiz/data/contents/test_content_categorize.py
tests/psiz/data/contents/test_content_rank.py
tests/psiz/data/contents/test_content_rate.py
tests/psiz/data/generators/test_weighted_rank.py
tests/psiz/data/groups/test_group.py
tests/psiz/data/outcomes/test_continuous.py
tests/psiz/data/outcomes/test_sparse_categorical.py
tests/psiz/keras/conftest.py
tests/psiz/keras/test_splitter.py
tests/psiz/keras/constraints/test_center.py
tests/psiz/keras/constraints/test_greater_than.py
tests/psiz/keras/constraints/test_less_equal_than.py
tests/psiz/keras/constraints/test_less_than.py
tests/psiz/keras/constraints/test_min_max.py
tests/psiz/keras/constraints/test_non_neg_norm.py
tests/psiz/keras/initializers/test_dirichlet.py
tests/psiz/keras/initializers/test_softplus_uniform.py
tests/psiz/keras/layers/conftest.py
tests/psiz/keras/layers/activations/test_exponential.py
tests/psiz/keras/layers/activations/test_heavy_tailed.py
tests/psiz/keras/layers/activations/test_inverse.py
tests/psiz/keras/layers/activations/test_students_t.py
tests/psiz/keras/layers/behaviors/test_alcove_cell.py
tests/psiz/keras/layers/behaviors/test_logistic.py
tests/psiz/keras/layers/behaviors/test_soft_rank.py
tests/psiz/keras/layers/embeddings/conftest.py
tests/psiz/keras/layers/embeddings/test_embedding_shared.py
tests/psiz/keras/layers/embeddings/test_embedding_variational.py
tests/psiz/keras/layers/embeddings/test_stochastic_embeddings.py
tests/psiz/keras/layers/gates/conftest.py
tests/psiz/keras/layers/gates/test_braid_gate.py
tests/psiz/keras/layers/gates/test_drop.py
tests/psiz/keras/layers/gates/test_gate_adapter.py
tests/psiz/keras/layers/proximities/test_cosine_similarity.py
tests/psiz/keras/layers/proximities/test_generalized_inner_product.py
tests/psiz/keras/layers/proximities/test_inner_product.py
tests/psiz/keras/layers/proximities/test_minkowski.py
tests/psiz/keras/layers/proximities/test_minkowski_stochastic.py
tests/psiz/keras/layers/proximities/test_minkowski_variational.py
tests/psiz/keras/models/conftest.py
tests/psiz/keras/models/test_alcove_model.py
tests/psiz/keras/models/test_mixed_model.py
tests/psiz/keras/models/test_proximity_model.py
tests/psiz/keras/models/test_rank_model.py
tests/psiz/keras/models/test_rate_model.py
tests/psiz/keras/models/test_stochastic_model.py
tests/psiz/keras/ops/test_expand_dim_repeat.py
tests/psiz/keras/ops/test_ig_categorical.py
tests/psiz/keras/ops/test_ig_model_categorical.py
tests/psiz/keras/regularizers/test_attention_entropy.py
tests/psiz/keras/regularizers/test_squeeze.py
tests/psiz/keras/regularizers/test_stimulus_normed_l1.py
tests/psiz/mplot/test_bvn_ellipse.py
tests/psiz/mplot/test_embedding_input_dimension.py
tests/psiz/mplot/test_embedding_output_dimension.py
tests/psiz/mplot/test_heatmap_embeddings.py
tests/psiz/tfp/test_unpack_mvn.py
tests/psiz/utils/test_affine_mvn.py
tests/psiz/utils/test_choice_wo_replace.py
tests/psiz/utils/test_m_prefer_n.py
tests/psiz/utils/test_pairwise_indices.py
tests/psiz/utils/test_procrustes.py
tests/psiz/utils/test_random_combinations.py
tests/psiz/utils/test_sort_based_mask.py
tests/psiz/utils/test_strat_group_kfold.py