pg-sui 0.2.3__py3-none-any.whl → 1.6.16a3__py3-none-any.whl

pg_sui-1.6.16a3.dist-info/METADATA

@@ -0,0 +1,292 @@
+Metadata-Version: 2.4
+Name: pg-sui
+Version: 1.6.16a3
+Summary: Python machine and deep learning API to impute missing genotypes
+Author-email: "Drs. Bradley T. Martin and Tyler K. Chafin" <evobio721@gmail.com>
+Maintainer-email: "Dr. Bradley T. Martin" <evobio721@gmail.com>
+License: GNU General Public License v3 (GPLv3)
+Project-URL: Homepage, https://github.com/btmartin721/PG-SUI
+Project-URL: Documentation, https://pg-sui.readthedocs.io/en/latest/
+Project-URL: Source, https://github.com/btmartin721/PG-SUI.git
+Project-URL: BugTracker, https://github.com/btmartin721/PG-SUI/issues
+Keywords: impute,imputation,AI,deep learning,machine learning,neural network,vae,autoencoder,ubp,nlpca,population genetics,unsupervised,supervised,bioinformatics,snp,genomics,genotype,missing data,data analysis,data science,statistics,data visualization,python
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Education
+Classifier: License :: OSI Approved :: GNU General Public License v3 (GPLv3)
+Classifier: Topic :: Scientific/Engineering :: Bio-Informatics
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Scientific/Engineering :: Information Analysis
+Classifier: Topic :: Scientific/Engineering :: Visualization
+Classifier: Operating System :: MacOS
+Classifier: Operating System :: MacOS :: MacOS X
+Classifier: Operating System :: Unix
+Classifier: Operating System :: POSIX
+Classifier: Natural Language :: English
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: matplotlib
+Requires-Dist: numpy>=2.1
+Requires-Dist: pandas>=2.2.2
+Requires-Dist: scikit-learn>=1.4
+Requires-Dist: scipy
+Requires-Dist: seaborn
+Requires-Dist: torch
+Requires-Dist: tqdm
+Requires-Dist: toytree
+Requires-Dist: optuna
+Requires-Dist: rich
+Requires-Dist: rich[jupyter]
+Requires-Dist: snpio
+Provides-Extra: intel
+Requires-Dist: scikit-learn-intelex; extra == "intel"
+Provides-Extra: docs
+Requires-Dist: sphinx; extra == "docs"
+Requires-Dist: sphinx-rtd-theme; extra == "docs"
+Requires-Dist: sphinx_autodoc_typehints; extra == "docs"
+Requires-Dist: sphinxcontrib-napoleon; extra == "docs"
+Requires-Dist: sphinxcontrib-programoutput; extra == "docs"
+Provides-Extra: dev
+Requires-Dist: twine; extra == "dev"
+Requires-Dist: wheel; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: sphinx; extra == "dev"
+Requires-Dist: sphinx-rtd-theme; extra == "dev"
+Requires-Dist: sphinx-autodoc-typehints; extra == "dev"
+Requires-Dist: sphinxcontrib-napoleon; extra == "dev"
+Requires-Dist: sphinxcontrib-programoutput; extra == "dev"
+Requires-Dist: requests; extra == "dev"
+Provides-Extra: optional
+Requires-Dist: PyObjC; extra == "optional"
+Provides-Extra: gui
+Requires-Dist: fastapi>=0.110; extra == "gui"
+Requires-Dist: uvicorn[standard]>=0.23; extra == "gui"
+Dynamic: license-file
+
+# PG-SUI
+
+![PG-SUI Logo](https://github.com/btmartin721/PG-SUI/blob/master/img/pgsui-logo-faded.png)
+
+Population Genomic Supervised and Unsupervised Imputation.
+
+## About PG-SUI
+
+PG-SUI is a Python 3 API that uses machine learning to impute missing values from population genomic SNP data. There are several supervised and unsupervised machine learning algorithms available to impute missing data, as well as some non-machine learning imputers that are useful.
+
+Below is some general information and a basic tutorial. For more detailed information, see our [API Documentation](https://pg-sui.readthedocs.io/en/latest/).
+
+### Unsupervised Imputation Methods
+
+Unsupervised imputers include three custom neural network models:
+
++ Variational Autoencoder (VAE) [1](#1)
+  + VAE models train themselves to reconstruct their input (i.e., the genotypes) [1](#1). To use VAE for imputation, the missing values are masked and the VAE model gets trained to reconstruct only on known values. Once the model is trained, it is then used to predict the missing values.
++ Autoencoder [2](#2)
+  + A standard autoencoder that trains the input to predict itself [2](#2). As with VAE, missing values are masked and the model gets trained only on known values. Predictions are then made on the missing values.
++ Non-linear Principal Component Analysis (NLPCA) [3](#3)
+  + NLPCA initializes random, reduced-dimensional input, then trains itself by using the known values (i.e., genotypes) as targets and refining the random input until it accurately predicts the genotype output [3](#3). The trained model can then predict the missing values.
++ Unsupervised Backpropagation (UBP) [4](#4)
+  + UBP is an extension of NLPCA that runs over three phases [4](#4). Phase 1 refines the randomly generated, reduced-dimensional input in a single layer perceptron neural network to obtain good initial input values. Phase 2 uses the refined reduced-dimensional input from phase 1 as input into a multi-layer perceptron (MLP), but in Phase 2 only the neural network weights are refined. Phase three uses an MLP to refine both the weights and the reduced-dimensional input. Once the model is trained, it can be used to predict the missing values.
+
+### Supervised Imputation Methods
+
+Supervised methods utilze the scikit-learn's ``IterativeImputer``, which is based on the MICE (Multivariate Imputation by Chained Equations) algorithm [5](#5), and iterates over each SNP site (i.e., feature) while uses the N nearest neighbor features to inform the imputation. The number of nearest features can be adjusted by users. IterativeImputer currently works with the following scikit-learn classifiers:
+
++ ImputeRandomForest
++ ImputeHistGradientBoosting
+
+See the [scikit-learn documentation](https://scikit-learn.org) for more information on IterativeImputer and each of the classifiers.
+
+### Non-Machine Learning (Deterministic) Methods
+
+We also include several deterministic options for imputing missing data, including:
+
++ Per-population mode per SNP site
++ Overall mode per SNP site
+
+## Installing PG-SUI
+
+PG-SUI supports both pip and conda distributions. Both are kept current with up-to-date releases.
+
+### Installation with Pip
+
+To install PG-SUI with pip, do the following. It is strongly recommended to install pg-sui in a virtual environment.
+
+``` shell
+python3 -m venv .pgsui-venv
+source .pgsui-venv/bin/activate
+pip install pg-sui
+```
+
+### Installation with Anaconda
+
+To install PG-SUI with Anaconda, do the following:
+
+``` shell
+conda create -n pgsui-env python=3.12
+conda activate pgsui-env
+conda install -c btmartin721 pg-sui
+```
+
+### Docker Container
+
+We also maintains a Docker image that comes with PG-SUI preinstalled. This can be useful for automated worklows such as Nextflow or Snakemake.
+
+``` shell
+docker pull pg-sui:latest
+```
+
+### Optional MacOS GUI
+
+PG-SUI ships an optional Electron GUI (Graphical User Interface) wrapper around the Python CLI. Currently for the GUI, only MacOS is supported.
+
+1. Install the Python-side extras (FastAPI/ uvicorn helper) if you want to serve from Python:
+   `pip install pg-sui[gui]`
+2. Install [Node.js](https://nodejs.org) and fetch the app dependencies:
+   `pgsui-gui-setup`
+3. Launch the graphical interface:
+   `pgsui-gui`
+
+The GUI shells out to the same CLI underneath, so presets, overrides, and YAML configs behave identically.
+
+## Input Data
+
+You can read your input files as a GenotypeData object from the [SNPio](https://snpio.readthedocs.io/en/latest/) package. SNPio supports the VCF, PHYLIP, STRUCTURE, and GENEPOP input file formats.
+
+``` python
+# Import snpio. Automatically installed with pg-sui.
+from snpio import VCFReader
+
+# Read in VCF alignment.
+# SNPio also supports PHYLIP, STRUCTURE, and GENEPOP input file formats.
+data = VCFReader(
+    filename="pgsui/example_data/phylogen_subset14K.vcf.gz,
+    popmapfile="pgsui/example_data/popmaps/phylogen_nomx.popmap", # optional
+    force_popmap=True, # optional
+)
+```
+
+## Supported Imputation Methods
+
+There are several supported algorithms PG-SUI uses to impute missing data. Each one can be run by calling the corresponding class. You must provide a GenotypeData instance as the first positional argument.
+
+You can import all the supported methods with the following:
+
+``` python
+from pgsui import ImputeUBP, ImputeVAE, ImputeNLPCA, ImputeAutoencoder, ImputeRefAllele, ImputeMostFrequent, ImputeRandomForest, ImputeHistGradientBoosting
+```
+
+### Unsupervised Imputers
+
+The four unsupervised imputers can be run by initializing them with the SNPio ``GenotypeData`` object and then calling ``fit()`` and ``transform()``.
+
+``` python
+# Initialize the models, then fit and impute
+vae = ImputeVAE(data) # Variational autoencoder
+vae.fit()
+vae_imputed = vae.transform()
+
+nlpca = ImputeNLPCA(data) # Nonlinear PCA
+nlpca.fit()
+nlpca_imputed = nlpca.transform()
+
+ubp = ImputeUBP(data) # Unsupervised backpropagation
+ubp.fit()
+ubp_imputed = ubp.transform()
+
+ae = ImputeAutoencoder(data) # standard autoencoder
+ae.fit()
+ae_imputed = ae.transform()
+```
+
+The ``*_imputed`` objects will be NumPy arrays that are compatible with SNPio's ``GenotypeData`` objects.
+
+### Supervised Imputers
+
+Various supervised imputation options are supported, and these use the same API design.
+
+``` python
+# Supervised IterativeImputer classifiers
+
+# Random Forest
+rf = ImputeRandomForest(data)
+rf.fit()
+imputed_rf = rf.transform()
+
+# HistGradientBoosting
+hgb = ImputeHistGradientBoosting(data)
+hgb.fit()
+imputed_hgb = hgb.transform()
+```
+
+### Non-machine learning methods
+
+The following deterministic methods are supported. ``ImputeMostFrequent`` supports the mode-per-population or overall (global) mode options to inform imputation.
+
+``` python
+# Per-population, per-locus mode
+pop_mode = ImputeMostFrequent(data, by_populations=True)
+pop_mode.fit()
+imputed_pop_mode = pop_mode.transform()
+
+# Per-locus mode
+mode = ImputeMostFrequent(data, by_populations=False)
+mode.fit()
+imputed_mode = mode.transform()
+```
+
+Or, always replace missing values with the reference allele.
+
+``` python
+ref = ImputeRefAllele(data)
+ref.fit()
+imputed_ref = ref.transform()
+```
+
+## Command-Line Interface
+
+Run the PG-SUI CLI with ``pg-sui`` (installed alongside the library). The CLI follows the same precedence model as the Python API:
+
+``code defaults  <  preset (--preset)  <  YAML (--config)  <  explicit CLI flags  <  --set key=value``.
+
+Recent releases add explicit switches for the simulated-missingness workflow shared by the neural and supervised models:
+
++ ``--sim-strategy`` selects one of ``random``, ``random_weighted``, ``random_weighted_inv``, ``nonrandom``, ``nonrandom_weighted``.
++ ``--sim-prop`` sets the proportion of observed calls to temporarily mask when building the evaluation set.
++ ``--simulate-missing`` disables simulated masking entirely (store-false flag); omit it to inherit preset/YAML defaults or re-enable via ``--set sim.simulate_missing=True``.
+
+Example:
+
+``` shell
+pg-sui \
+  --vcf data.vcf.gz \
+  --popmap pops.popmap \
+  --models ImputeUBP ImputeVAE \
+  --preset balanced \
+  --sim-strategy random_weighted_inv \
+  --sim-prop 0.25 \
+  --prefix ubp_and_vae \
+  --n-jobs 4 \
+  --tune-n-trials 100 \
+  --set tune.enabled=True
+```
+
+CLI overrides cascade into every selected model, so a single invocation can evaluate multiple imputers with a consistent simulation strategy and output prefix.
+
+## References
+
+1. Kingma, D.P. & Welling, M. (2013). Auto-encoding variational bayes. In: Proceedings  of  the  International Conference on Learning Representations (ICLR). arXiv:1312.6114 [stat.ML].
+
+2. Hinton, G.E., & Salakhutdinov, R.R. (2006). Reducing the dimensionality of data with neural networks. Science, 313(5786), 504-507.
+
+3. Scholz, M., Kaplan, F., Guy, C. L., Kopka, J., & Selbig, J. (2005). Non-linear PCA: a missing data approach. Bioinformatics, 21(20), 3887-3895.
+
+4. Gashler, M. S., Smith, M. R., Morris, R., & Martinez, T. (2016). Missing value imputation with unsupervised backpropagation. Computational Intelligence, 32(2), 196-215.
+
+5. Stef van Buuren, Karin Groothuis-Oudshoorn (2011). mice: Multivariate Imputation by Chained Equations in R. Journal of Statistical Software 45: 1-67.

pg_sui-1.6.16a3.dist-info/RECORD

@@ -0,0 +1,81 @@
+pg_sui-1.6.16a3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+pgsui/__init__.py,sha256=wQFzVX6vh8aUva1LCvP42jS7rcKCpkaU52YfZIy61q8,1493
+pgsui/_version.py,sha256=Ob9Wzde013P9zPCsc2eaS--ZJQaRFPqutqIvBNJDnP4,714
+pgsui/cli.py,sha256=F6v7Bv073NgH3Ku_Lb6DSciTCLzM_o49ktrKnjkiHJg,30229
+pgsui/data_processing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/data_processing/config.py,sha256=g5G7pjWG4uU2BRvBu_DpO0J_4X1Foa11X69imBWjaKA,20483
+pgsui/data_processing/containers.py,sha256=mtVExBEQQXCID_TbngxY23I690d09WIzmJBdVa7HIo0,51513
+pgsui/data_processing/transformers.py,sha256=kdwOTmfDjgQ3RmiwQIK7LYL4vQUpgA4bob7NHYgnYRM,30998
+pgsui/electron/bootstrap.py,sha256=wnrXgX-hiqrMMFE9WGoD-UC8zeK2ZP6Kupu68PodVWI,1185
+pgsui/electron/launch.py,sha256=M60o_jub77kJL-B9d_sMB7LYuTzWlOnQXR09efmCX2o,1715
+pgsui/electron/package.json,sha256=12hbBq7xincW5V4645TTC58jfkA2rPgFP_eLb_WbhKo,372
+pgsui/electron/app/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/electron/app/__main__.py,sha256=rF7tIpuy4OvMnzLms-fAUNfqYINEyR0Iofh1KKu5RI0,103
+pgsui/electron/app/main.js,sha256=yobpq13uOdoLzwonCxjAFECwqop0bk8_-7pYBGci04U,8252
+pgsui/electron/app/package-lock.json,sha256=0sp5XM4tclppG4vqFDtpKDIV9dBC0JVtjBtYKpSXIP8,246701
+pgsui/electron/app/package.json,sha256=Mk5FALnz7hlV_5r5f84HgYp2oLjVD9SHnXmMVrTfzmo,1276
+pgsui/electron/app/preload.js,sha256=L3lPsTGfgiGXLmQFwktW11En1YSb01hWlJD6kzzmbhE,803
+pgsui/electron/app/server.py,sha256=q8gWwgkbaDjNZTT_dR1HXm45xv4V7wjrPk-Uc6axros,4568
+pgsui/electron/app/extra-resources/.gitkeep,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+pgsui/electron/app/icons/icons/1024x1024.png,sha256=psZk62iG308QN9XP7sDCifGAHwG7__W-e2EUrF9hss0,42270
+pgsui/electron/app/icons/icons/128x128.png,sha256=R9Za6fj289L432EdwAX7pUj4p4qy7iTMzghrrQzocCg,4705
+pgsui/electron/app/icons/icons/16x16.png,sha256=TaVFFA8V4LQX_KwZCjko_eRPibmYEy1JSkuPicUZ0oQ,704
+pgsui/electron/app/icons/icons/24x24.png,sha256=vMmSBmnTPhaVBfV_oCK_r3wykBq9JneyQu4TlHUVjBU,1094
+pgsui/electron/app/icons/icons/256x256.png,sha256=D97Y8KYOg_RcsR8pNDcxyxtRz6X3qKkKw-oChy29tuY,9044
+pgsui/electron/app/icons/icons/32x32.png,sha256=ah-Z-WybnXG801UV2xwAdvUNmQmy1C29EWLNlTgN8kI,1355
+pgsui/electron/app/icons/icons/48x48.png,sha256=WYVea1_leKVpowEQCrbmV1zG5aMBRh-aMKpiziTKWYc,1995
+pgsui/electron/app/icons/icons/512x512.png,sha256=A1YP6sKNFZEdVl7vVlIAUjunSzFhtXS0_e7bxRiadOQ,17784
+pgsui/electron/app/icons/icons/64x64.png,sha256=tH4qDy3My1DPZz4QzSyz4P6_eel_3t246xKhtm1xKwg,2610
+pgsui/electron/app/icons/icons/icon.icns,sha256=qaJExjDDJQk-uSbTIyPFn6a1hb8JRxs5HQmYdzZdb0c,108937
+pgsui/electron/app/icons/icons/icon.ico,sha256=ljjD98wBOVfUSV3gPlMzXZ4FAprRdrAQENCZ76KTFmo,361102
+pgsui/electron/app/ui/logo.png,sha256=uHhtVMYQsSCkcpeS8JPo-DVsAqwg3G4t-0lYuT4AP6E,38243
+pgsui/electron/app/ui/renderer.js,sha256=SzG7WhiQ3lDp2DKgkOnRO8lohAyhXlnQi8V-r7lILeI,5351
+pgsui/electron/app/ui/styles.css,sha256=NQPGczGXxw8XjrLbqKZtEQD-SICvciFuUfgtKm_FuHk,2675
+pgsui/electron/app/ui/ui_shim.js,sha256=4qu_9qpMc_p-Boih0o8I2OAVL9TMS4DUUk7BRQIfCeM,2513
+pgsui/example_data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/example_data/phylip_files/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/example_data/phylip_files/test.phy,sha256=LEq2Q3hjmPVDP2uF1Ai8OH2If5Y58ny-GRfuXa3l61c,238537
+pgsui/example_data/popmaps/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/example_data/popmaps/phylogen_nomx.popmap,sha256=JR84kJ5jVjJ45k4kpvGVt2JNJHApQg1xNs2pKPt-V4k,3310
+pgsui/example_data/structure_files/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/example_data/structure_files/test.pops.2row.allsites.str,sha256=fbmLcOUQSKyy71A77NPtJJMLSOkquPky0WiPlV4qHhA,967878
+pgsui/example_data/vcf_files/phylogen_subset14K.vcf.gz,sha256=B9sxdIGPSbw4m4MTX_B1VHDx5cDicMNn83AFJy7SEAU,8053540
+pgsui/example_data/vcf_files/phylogen_subset14K.vcf.gz.tbi,sha256=0gHDvboDAEZLQbXdYfUOiJL3oDOr1mOlaQxdlcz_erg,106087
+pgsui/impute/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/impute/deterministic/imputers/allele_freq.py,sha256=tqaMggiNs6hxs4CN3No2d5llmmIPt0jQhHT0mHry2IM,29573
+pgsui/impute/deterministic/imputers/mode.py,sha256=lQmrOYEyfQidqGaG86QB5HeM66IeOFLK1UoCvh-iDno,35049
+pgsui/impute/deterministic/imputers/nmf.py,sha256=171_TTDZAe1NFjbmKQTOlPDe_rA1aO8q2Th5z0w2RU8,8086
+pgsui/impute/deterministic/imputers/phylo.py,sha256=uN86_L2TeiUFOZLdU8pwymRiQf0SI7Sl6SAnCbAywVQ,38873
+pgsui/impute/deterministic/imputers/ref_allele.py,sha256=OGdv4n1avIArItZ-V0kqX2g4OOZc2felh7HM1ikY2iU,28075
+pgsui/impute/supervised/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/impute/supervised/base.py,sha256=A92x1pS8DO0OwbhQem3KBAMbWs368KJcVl88zJ7wE10,13756
+pgsui/impute/supervised/imputers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/impute/supervised/imputers/hist_gradient_boosting.py,sha256=5LZsee2R9tbshmgVvLDIjGDihiQPvp9XfbaGwzL35E8,11509
+pgsui/impute/supervised/imputers/random_forest.py,sha256=jpeaLjhzYrwRPc9nPasLgwOqBoHlBlymHMc3x5OkBWA,10393
+pgsui/impute/unsupervised/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/impute/unsupervised/base.py,sha256=DqgP8QNKmbUaljzx_lWC_zFuWTUO3iiRseYTQEWNG0Q,46945
+pgsui/impute/unsupervised/callbacks.py,sha256=jkxncpKNRdssImv4N5c-Hq8VcA07QvxLoct7EqDW9RE,5026
+pgsui/impute/unsupervised/loss_functions.py,sha256=f18uQnerj0KF9xcU2I1_Y2OCKKguXXaEDaYhJg0XElY,10089
+pgsui/impute/unsupervised/nn_scorers.py,sha256=-rl5MBJm2GN6E1wPBIe1wMgdrHEHhYooUUxVbfcf1Z8,9758
+pgsui/impute/unsupervised/imputers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/impute/unsupervised/imputers/autoencoder.py,sha256=k8GJ80hysAteT-QUmRRtWfPcvH-QjUFHrWBZGVuFH1c,57872
+pgsui/impute/unsupervised/imputers/nlpca.py,sha256=AChGz0Gp9Yt0qF8vtrDI4t5iIr70LQRb-8pFGQBYLZo,69086
+pgsui/impute/unsupervised/imputers/ubp.py,sha256=sBJ9VSW2tTzXVjnXogjhuj6GixRaH0HcQ2B7cK4y6tw,71361
+pgsui/impute/unsupervised/imputers/vae.py,sha256=hK3ZOxyLjQqm-3Hcq2BMUZOghsJiDGYNRQjqA7zoncs,54713
+pgsui/impute/unsupervised/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/impute/unsupervised/models/autoencoder_model.py,sha256=mHmfTkldJNpN7Dy7RTS2RnkE2L_K1rChNjpjDvzRlEQ,12333
+pgsui/impute/unsupervised/models/nlpca_model.py,sha256=1NathvhsirBtd9UcmeJzRoVf7oi7PfDmRpt18Di63Cg,8021
+pgsui/impute/unsupervised/models/ubp_model.py,sha256=4guGkQzCTIsDnImOjJV5kG1xc1ST9oO4aUeXrSBSpQg,8491
+pgsui/impute/unsupervised/models/vae_model.py,sha256=hMG7K1OR95qLEEcprGSzIoTIISXHSu2yWDy6QkG93Lg,15576
+pgsui/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pgsui/utils/classification_viz.py,sha256=WkGQkEIkpTMLQvGdGoU53kn1iNUO9ipeg5stSY1rcFg,22012
+pgsui/utils/logging_utils.py,sha256=o_ElRL05B_DrbALYkuW8s_azfKQiN8kJ4oXwshyIMyI,521
+pgsui/utils/misc.py,sha256=Mw5CsspFJkDAcCRufk-lO7fKyVoYK7PRYXkLXKswUjI,3065
+pgsui/utils/plotting.py,sha256=d5CTzGIpanu3j6rEB6fq_F1g8w_A2Ti_XiedRjIFFII,42444
+pgsui/utils/pretty_metrics.py,sha256=dtN7Ohcx3qJYCw4JeJCXvthGDdSV7bgE8v6EGwHSAE0,9862
+pgsui/utils/scorers.py,sha256=sL2upL2ZZMFBTMM4DiGiWeXrqc_fp1RRbleYCnuRUhw,12564
+pg_sui-1.6.16a3.dist-info/METADATA,sha256=G87-keKhu56QhKFiAeGQVSD36EH_W2uNf9u4wciAsmU,12320
+pg_sui-1.6.16a3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pg_sui-1.6.16a3.dist-info/entry_points.txt,sha256=xidyl6yqQv7oj3XSzZC6Vv9l7aNgbHi_pjv-dJjGJds,129
+pg_sui-1.6.16a3.dist-info/top_level.txt,sha256=87-oDpfY6sDY_uN-OM2lcnrgPesifhzwqFOajp9ukz0,6
+pg_sui-1.6.16a3.dist-info/RECORD,,

{pg_sui-0.2.3.dist-info → pg_sui-1.6.16a3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.41.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

pg_sui-1.6.16a3.dist-info/entry_points.txt

@@ -0,0 +1,4 @@
+[console_scripts]
+pg-sui = pgsui.cli:main
+pgsui-gui = pgsui.electron.launch:main
+pgsui-gui-setup = pgsui.electron.bootstrap:main

pg_sui-1.6.16a3.dist-info/top_level.txt

@@ -0,0 +1 @@
+pgsui

pgsui/__init__.py

@@ -1,63 +1,44 @@
 ## PG-SUI package by Bradley T. Martin and Tyler K. Chafin
 ## E-mail: evobio721@gmail.com
-## Version 0.1, completed 13-Dec-2021
-
-# Suppresses tensorflow GPU warnings.
-import os
-import warnings
-
-os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
-
-from pgsui.utils.misc import get_processor_name
-
-
-# Requires scikit-learn-intellex package
-if get_processor_name().strip().startswith("Intel"):
-    try:
-        from sklearnex import patch_sklearn
-
-        patch_sklearn()
-        intelex = True
-    except (ImportError, TypeError):
-        warnings.warn(
-            "Intel CPU detected but scikit-learn-intelex is not installed. We recommend installing it to speed up computation if your hardware supports it."
-        )
-        intelex = False
-else:
-    intelex = False
-
-os.environ["intelex"] = str(intelex)
-
-from pgsui.impute.estimators import (
-    ImputeKNN,
-    ImputeNLPCA,
-    ImputeRandomForest,
-    ImputeStandardAutoEncoder,
-    ImputeUBP,
-    ImputeVAE,
-    ImputeXGBoost,
+from ._version import version as __version__
+
+from pgsui.data_processing.containers import (
+    AutoencoderConfig,
+    HGBConfig,
+    MostFrequentConfig,
+    NLPCAConfig,
+    RefAlleleConfig,
+    RFConfig,
+    UBPConfig,
+    VAEConfig,
 )
-
-from pgsui.impute.simple_imputers import (
-    ImputePhylo,
-    ImputeMF,
-    ImputeAlleleFreq,
-    ImputeRefAllele,
+from pgsui.impute.deterministic.imputers.mode import ImputeMostFrequent
+from pgsui.impute.deterministic.imputers.ref_allele import ImputeRefAllele
+from pgsui.impute.supervised.imputers.hist_gradient_boosting import (
+    ImputeHistGradientBoosting,
 )
-
-from pgsui.data_processing.transformers import SimGenotypeDataTransformer
+from pgsui.impute.supervised.imputers.random_forest import ImputeRandomForest
+from pgsui.impute.unsupervised.imputers.autoencoder import ImputeAutoencoder
+from pgsui.impute.unsupervised.imputers.nlpca import ImputeNLPCA
+from pgsui.impute.unsupervised.imputers.ubp import ImputeUBP
+from pgsui.impute.unsupervised.imputers.vae import ImputeVAE
 
 __all__ = [
-    "ImputeUBP",
+    "ImputeAutoencoder",  # Unsupervised imputer classes
     "ImputeVAE",
-    "ImputeXGBoost",
-    "ImputeStandardAutoEncoder",
-    "ImputeRandomForest",
     "ImputeNLPCA",
-    "ImputeKNN",
-    "SimGenotypeDataTransformer",
-    "ImputePhylo",
-    "ImputeMF",
-    "ImputeAlleleFreq",
-    "ImputeRefAllele",
+    "ImputeUBP",
+    "ImputeRandomForest",  # Supervised imputer classes
+    "ImputeHistGradientBoosting",
+    "ImputeRefAllele",  # Deterministic imputer classes
+    "ImputeMostFrequent",
+    "AutoencoderConfig",  # Unsupervised imputer configs
+    "VAEConfig",
+    "NLPCAConfig",
+    "UBPConfig",
+    "MostFrequentConfig",  # Deterministic imputer configs
+    "RefAlleleConfig",
+    "RFConfig",  # Supervised imputer configs
+    "HGBConfig",
+    "__version__",
 ]

pgsui/_version.py

@@ -0,0 +1,34 @@
+# file generated by setuptools-scm
+# don't change, don't track in version control
+
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
+
+TYPE_CHECKING = False
+if TYPE_CHECKING:
+    from typing import Tuple
+    from typing import Union
+
+    VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
+else:
+    VERSION_TUPLE = object
+    COMMIT_ID = object
+
+version: str
+__version__: str
+__version_tuple__: VERSION_TUPLE
+version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
+
+__version__ = version = '1.6.16a3'
+__version_tuple__ = version_tuple = (1, 6, 16, 'a3')
+
+__commit_id__ = commit_id = None