Create register function for PySpark #17
Closed
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
karenfeng
force-pushed
the
register-sql-extensions
branch
from
a2e463b
to
f59948a
Compare
henrydavidge
pushed a commit
to henrydavidge/glow
that referenced
this pull request
Jun 22, 2020
Signed-off-by: Karen Feng <[email protected]>
henrydavidge
pushed a commit
to henrydavidge/glow
that referenced
this pull request
Jun 22, 2020
Signed-off-by: Henry Davidge <[email protected]>
henrydavidge
pushed a commit
to henrydavidge/glow
that referenced
this pull request
Jun 22, 2020
Signed-off-by: Karen Feng <[email protected]> Signed-off-by: Henry Davidge <[email protected]>
henrydavidge
added a commit
that referenced
this pull request
Jun 22, 2020
* Add Leland's demo notebook * block_variants_and_samples Transformer to create genotype DataFrame for WGR (#2) * blocks Signed-off-by: kianfar77 <[email protected]> * test vcf Signed-off-by: kianfar77 <[email protected]> * transformer Signed-off-by: kianfar77 <[email protected]> * remove extra Signed-off-by: kianfar77 <[email protected]> * refactor and conform with ridge namings Signed-off-by: kianfar77 <[email protected]> * test Signed-off-by: kianfar77 <[email protected]> * test files Signed-off-by: kianfar77 <[email protected]> * remove extra file Signed-off-by: kianfar77 <[email protected]> * sort_key Signed-off-by: kianfar77 <[email protected]> * feat: ridge models for wgr added (#1) * feat: ridge models for wgr added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Doc strings added for levels/functions.py Some typos fixed in ridge_model.py Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * ridge_model and RidgeReducer unit tests added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * RidgeRegression unit tests added test data README added ridge_udfs.py docstrings added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Changes made to accessing the sample ID map and more docstrings The map_normal_eqn and score_models functions previously expected the sample IDs for a given sample block to be found in the Pandas DataFrame, which mean we had to join them on before the .groupBy().apply(). These functions now expect the sample block to sample IDs mapping to be provided separately as a dict, so that the join is no longer required. RidgeReducer and RidgeRegression APIs remain unchanged. docstrings have been added for RidgeReducer and RidgeRegression classes. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Refactored object names and comments to reflect new terminology Where 'block' was previously used to refer to the set of columns in a block, we now use 'header_block' Where 'group' was previously used to refer to the set of samples in a block, we now use 'sample_block' Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * [HLS-539] Fix compatibility between blocked GT transformer and WGR (#6) * WIP Signed-off-by: Karen Feng <[email protected]> * existing tests pass Signed-off-by: Karen Feng <[email protected]> * rename file Signed-off-by: Karen Feng <[email protected]> * Add compat test Signed-off-by: Karen Feng <[email protected]> * scalafmt Signed-off-by: Karen Feng <[email protected]> * collect minimal columns Signed-off-by: Karen Feng <[email protected]> * address comments Signed-off-by: Karen Feng <[email protected]> * Test fixup Signed-off-by: Karen Feng <[email protected]> * Spark 3 needs more recent PyArrow, reduce mem consumption by removing unnecessary caching Signed-off-by: Karen Feng <[email protected]> * PyArrow 0.15.1 only with PySpark 3 Signed-off-by: Karen Feng <[email protected]> * Don't use toPandas() Signed-off-by: Karen Feng <[email protected]> * Upgrade pyarrow Signed-off-by: Karen Feng <[email protected]> * Only register once Signed-off-by: Karen Feng <[email protected]> * Minimize memory usage Signed-off-by: Karen Feng <[email protected]> * Select before head Signed-off-by: Karen Feng <[email protected]> * set up/tear down Signed-off-by: Karen Feng <[email protected]> * Try limiting pyspark memory Signed-off-by: Karen Feng <[email protected]> * No teardown Signed-off-by: Karen Feng <[email protected]> * Extend timeout Signed-off-by: Karen Feng <[email protected]> * Simplify ordering logic in levels code (#7) * WIP Signed-off-by: Karen Feng <[email protected]> * existing tests pass Signed-off-by: Karen Feng <[email protected]> * rename file Signed-off-by: Karen Feng <[email protected]> * Add compat test Signed-off-by: Karen Feng <[email protected]> * scalafmt Signed-off-by: Karen Feng <[email protected]> * collect minimal columns Signed-off-by: Karen Feng <[email protected]> * start changing for readability * use input label ordering * rename create_row_indexer * undo column sort * change reduce Signed-off-by: Henry D <[email protected]> * further simplify reduce * sorted alpha names * remove ordering * comments Signed-off-by: Henry D <[email protected]> * Set arrow env var in build Signed-off-by: Henry D <[email protected]> * faster sort * add test file * undo test data change * >= * formatting * empty Co-authored-by: Karen Feng <[email protected]> * Limit Spark memory conf in tests (#9) * yapf Signed-off-by: Karen Feng <[email protected]> * yapf transform Signed-off-by: Karen Feng <[email protected]> * Set driver memory Signed-off-by: Karen Feng <[email protected]> * Try changing spark mem Signed-off-by: Karen Feng <[email protected]> * match java tests Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * remove driver memory flag Signed-off-by: Karen Feng <[email protected]> * Improve partitioning in block_variants_and_samples transformer (#11) Signed-off-by: kianfar77 <[email protected]> * Remove unnecessary header_block grouping (#10) * cleanup Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * cleanup Signed-off-by: Karen Feng <[email protected]> * Create sample ID blocking helper functions (#12) * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * tests Signed-off-by: Karen Feng <[email protected]> * simplify tests Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * yapf Signed-off-by: Karen Feng <[email protected]> * index map compat Signed-off-by: Karen Feng <[email protected]> * Add docs Signed-off-by: Karen Feng <[email protected]> * Add more tests Signed-off-by: Karen Feng <[email protected]> * pass args as ints Signed-off-by: Karen Feng <[email protected]> * Don't roll our own splitter Signed-off-by: Karen Feng <[email protected]> * rename sample_index to sample_blocks Signed-off-by: Karen Feng <[email protected]> * Add type-checking to WGR APIs (#14) * Add type-checking to APIs Signed-off-by: Karen Feng <[email protected]> * Check valid alphas Signed-off-by: Karen Feng <[email protected]> * check 0 sig Signed-off-by: Karen Feng <[email protected]> * Add to install_requires list Signed-off-by: Karen Feng <[email protected]> * cleanup comments Signed-off-by: Karen Feng <[email protected]> * Add covariate support (#13) * Added necessary modifications to accomodate covariates in model fitting. The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X. This PR makes numerous changes to accomodate covariate matrix C. Adding covariates required the following breaking changes to the APIs: * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform(): * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf) * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf) Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument. Two new tests have been added to test_ridge_regression.py to test run modes with covariates: * test_ridge_reducer_transform_with_cov * test_two_level_regression_with_cov Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Cleaned up one unnecessary Pandas import Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Small changes for clarity and consistence with the rest of the code. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Forgot one usage of coalesce Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Added a couple of comments to explain logic and replaced usages of .values with .array Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Fixed one instance of the change .values -> .array where it was made in error. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Typo in test_ridge_regression.py. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Style auto-updates with yapfAll Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> Co-authored-by: Leland Barnard <[email protected]> Co-authored-by: Karen Feng <[email protected]> * Flatten estimated phenotypes (#15) * WIP Signed-off-by: Karen Feng <[email protected]> * Clean up tests Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * Order to match labeldf Signed-off-by: Karen Feng <[email protected]> * Check we tie-break Signed-off-by: Karen Feng <[email protected]> * cleanup Signed-off-by: Karen Feng <[email protected]> * tests Signed-off-by: Karen Feng <[email protected]> * test var name Signed-off-by: Karen Feng <[email protected]> * clean up tests Signed-off-by: Karen Feng <[email protected]> * Clean up docs Signed-off-by: Karen Feng <[email protected]> * Add fit_transform function to models (#17) Signed-off-by: Karen Feng <[email protected]> * Rename levels (#20) * Rename levels to wgr Signed-off-by: Karen Feng <[email protected]> * rename test files Signed-off-by: Karen Feng <[email protected]> * Add license headers (#21) * headers * executable * fix template rendering * yapf * add header to template * add header to template Signed-off-by: Henry D <[email protected]> Co-authored-by: Kiavash Kianfar <[email protected]> Co-authored-by: Karen Feng <[email protected]> Co-authored-by: Leland <[email protected]> Co-authored-by: Leland Barnard <[email protected]>
henrydavidge
added a commit
that referenced
this pull request
Jun 22, 2020
* Add Leland's demo notebook * block_variants_and_samples Transformer to create genotype DataFrame for WGR (#2) * blocks Signed-off-by: kianfar77 <[email protected]> * test vcf Signed-off-by: kianfar77 <[email protected]> * transformer Signed-off-by: kianfar77 <[email protected]> * remove extra Signed-off-by: kianfar77 <[email protected]> * refactor and conform with ridge namings Signed-off-by: kianfar77 <[email protected]> * test Signed-off-by: kianfar77 <[email protected]> * test files Signed-off-by: kianfar77 <[email protected]> * remove extra file Signed-off-by: kianfar77 <[email protected]> * sort_key Signed-off-by: kianfar77 <[email protected]> * feat: ridge models for wgr added (#1) * feat: ridge models for wgr added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Doc strings added for levels/functions.py Some typos fixed in ridge_model.py Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * ridge_model and RidgeReducer unit tests added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * RidgeRegression unit tests added test data README added ridge_udfs.py docstrings added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Changes made to accessing the sample ID map and more docstrings The map_normal_eqn and score_models functions previously expected the sample IDs for a given sample block to be found in the Pandas DataFrame, which mean we had to join them on before the .groupBy().apply(). These functions now expect the sample block to sample IDs mapping to be provided separately as a dict, so that the join is no longer required. RidgeReducer and RidgeRegression APIs remain unchanged. docstrings have been added for RidgeReducer and RidgeRegression classes. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Refactored object names and comments to reflect new terminology Where 'block' was previously used to refer to the set of columns in a block, we now use 'header_block' Where 'group' was previously used to refer to the set of samples in a block, we now use 'sample_block' Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * [HLS-539] Fix compatibility between blocked GT transformer and WGR (#6) * WIP Signed-off-by: Karen Feng <[email protected]> * existing tests pass Signed-off-by: Karen Feng <[email protected]> * rename file Signed-off-by: Karen Feng <[email protected]> * Add compat test Signed-off-by: Karen Feng <[email protected]> * scalafmt Signed-off-by: Karen Feng <[email protected]> * collect minimal columns Signed-off-by: Karen Feng <[email protected]> * address comments Signed-off-by: Karen Feng <[email protected]> * Test fixup Signed-off-by: Karen Feng <[email protected]> * Spark 3 needs more recent PyArrow, reduce mem consumption by removing unnecessary caching Signed-off-by: Karen Feng <[email protected]> * PyArrow 0.15.1 only with PySpark 3 Signed-off-by: Karen Feng <[email protected]> * Don't use toPandas() Signed-off-by: Karen Feng <[email protected]> * Upgrade pyarrow Signed-off-by: Karen Feng <[email protected]> * Only register once Signed-off-by: Karen Feng <[email protected]> * Minimize memory usage Signed-off-by: Karen Feng <[email protected]> * Select before head Signed-off-by: Karen Feng <[email protected]> * set up/tear down Signed-off-by: Karen Feng <[email protected]> * Try limiting pyspark memory Signed-off-by: Karen Feng <[email protected]> * No teardown Signed-off-by: Karen Feng <[email protected]> * Extend timeout Signed-off-by: Karen Feng <[email protected]> * Simplify ordering logic in levels code (#7) * WIP Signed-off-by: Karen Feng <[email protected]> * existing tests pass Signed-off-by: Karen Feng <[email protected]> * rename file Signed-off-by: Karen Feng <[email protected]> * Add compat test Signed-off-by: Karen Feng <[email protected]> * scalafmt Signed-off-by: Karen Feng <[email protected]> * collect minimal columns Signed-off-by: Karen Feng <[email protected]> * start changing for readability * use input label ordering * rename create_row_indexer * undo column sort * change reduce Signed-off-by: Henry D <[email protected]> * further simplify reduce * sorted alpha names * remove ordering * comments Signed-off-by: Henry D <[email protected]> * Set arrow env var in build Signed-off-by: Henry D <[email protected]> * faster sort * add test file * undo test data change * >= * formatting * empty Co-authored-by: Karen Feng <[email protected]> * Limit Spark memory conf in tests (#9) * yapf Signed-off-by: Karen Feng <[email protected]> * yapf transform Signed-off-by: Karen Feng <[email protected]> * Set driver memory Signed-off-by: Karen Feng <[email protected]> * Try changing spark mem Signed-off-by: Karen Feng <[email protected]> * match java tests Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * remove driver memory flag Signed-off-by: Karen Feng <[email protected]> * Improve partitioning in block_variants_and_samples transformer (#11) Signed-off-by: kianfar77 <[email protected]> * Remove unnecessary header_block grouping (#10) * cleanup Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * cleanup Signed-off-by: Karen Feng <[email protected]> * Create sample ID blocking helper functions (#12) * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * tests Signed-off-by: Karen Feng <[email protected]> * simplify tests Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * yapf Signed-off-by: Karen Feng <[email protected]> * index map compat Signed-off-by: Karen Feng <[email protected]> * Add docs Signed-off-by: Karen Feng <[email protected]> * Add more tests Signed-off-by: Karen Feng <[email protected]> * pass args as ints Signed-off-by: Karen Feng <[email protected]> * Don't roll our own splitter Signed-off-by: Karen Feng <[email protected]> * rename sample_index to sample_blocks Signed-off-by: Karen Feng <[email protected]> * Add type-checking to WGR APIs (#14) * Add type-checking to APIs Signed-off-by: Karen Feng <[email protected]> * Check valid alphas Signed-off-by: Karen Feng <[email protected]> * check 0 sig Signed-off-by: Karen Feng <[email protected]> * Add to install_requires list Signed-off-by: Karen Feng <[email protected]> * cleanup comments Signed-off-by: Karen Feng <[email protected]> * Add covariate support (#13) * Added necessary modifications to accomodate covariates in model fitting. The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X. This PR makes numerous changes to accomodate covariate matrix C. Adding covariates required the following breaking changes to the APIs: * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform(): * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf) * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf) Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument. Two new tests have been added to test_ridge_regression.py to test run modes with covariates: * test_ridge_reducer_transform_with_cov * test_two_level_regression_with_cov Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Cleaned up one unnecessary Pandas import Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Small changes for clarity and consistence with the rest of the code. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Forgot one usage of coalesce Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Added a couple of comments to explain logic and replaced usages of .values with .array Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Fixed one instance of the change .values -> .array where it was made in error. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Typo in test_ridge_regression.py. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Style auto-updates with yapfAll Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> Co-authored-by: Leland Barnard <[email protected]> Co-authored-by: Karen Feng <[email protected]> * Flatten estimated phenotypes (#15) * WIP Signed-off-by: Karen Feng <[email protected]> * Clean up tests Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * Order to match labeldf Signed-off-by: Karen Feng <[email protected]> * Check we tie-break Signed-off-by: Karen Feng <[email protected]> * cleanup Signed-off-by: Karen Feng <[email protected]> * tests Signed-off-by: Karen Feng <[email protected]> * test var name Signed-off-by: Karen Feng <[email protected]> * clean up tests Signed-off-by: Karen Feng <[email protected]> * Clean up docs Signed-off-by: Karen Feng <[email protected]> * Add fit_transform function to models (#17) Signed-off-by: Karen Feng <[email protected]> * support alpha inference Signed-off-by: Karen Feng <[email protected]> * test fixup Signed-off-by: Karen Feng <[email protected]> * more test fixup Signed-off-by: Karen Feng <[email protected]> * test fixups Signed-off-by: Karen Feng <[email protected]> * sub-sample Signed-off-by: Karen Feng <[email protected]> * test fixup Signed-off-by: Karen Feng <[email protected]> * address comments - only infer alphas during fit Signed-off-by: Karen Feng <[email protected]> * exception varies Signed-off-by: Karen Feng <[email protected]> * Rename levels (#20) * Rename levels to wgr Signed-off-by: Karen Feng <[email protected]> * rename test files Signed-off-by: Karen Feng <[email protected]> * Errors vary by Spark version Signed-off-by: Karen Feng <[email protected]> * Add license headers (#21) * headers * executable * fix template rendering * yapf Co-authored-by: Kiavash Kianfar <[email protected]> Co-authored-by: Karen Feng <[email protected]> Co-authored-by: Leland <[email protected]> Co-authored-by: Leland Barnard <[email protected]>
karenfeng
added a commit
that referenced
this pull request
Jun 23, 2020
* Add Leland's demo notebook * block_variants_and_samples Transformer to create genotype DataFrame for WGR (#2) * blocks Signed-off-by: kianfar77 <[email protected]> * test vcf Signed-off-by: kianfar77 <[email protected]> * transformer Signed-off-by: kianfar77 <[email protected]> * remove extra Signed-off-by: kianfar77 <[email protected]> * refactor and conform with ridge namings Signed-off-by: kianfar77 <[email protected]> * test Signed-off-by: kianfar77 <[email protected]> * test files Signed-off-by: kianfar77 <[email protected]> * remove extra file Signed-off-by: kianfar77 <[email protected]> * sort_key Signed-off-by: kianfar77 <[email protected]> * feat: ridge models for wgr added (#1) * feat: ridge models for wgr added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Doc strings added for levels/functions.py Some typos fixed in ridge_model.py Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * ridge_model and RidgeReducer unit tests added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * RidgeRegression unit tests added test data README added ridge_udfs.py docstrings added Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Changes made to accessing the sample ID map and more docstrings The map_normal_eqn and score_models functions previously expected the sample IDs for a given sample block to be found in the Pandas DataFrame, which mean we had to join them on before the .groupBy().apply(). These functions now expect the sample block to sample IDs mapping to be provided separately as a dict, so that the join is no longer required. RidgeReducer and RidgeRegression APIs remain unchanged. docstrings have been added for RidgeReducer and RidgeRegression classes. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Refactored object names and comments to reflect new terminology Where 'block' was previously used to refer to the set of columns in a block, we now use 'header_block' Where 'group' was previously used to refer to the set of samples in a block, we now use 'sample_block' Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * [HLS-539] Fix compatibility between blocked GT transformer and WGR (#6) * WIP Signed-off-by: Karen Feng <[email protected]> * existing tests pass Signed-off-by: Karen Feng <[email protected]> * rename file Signed-off-by: Karen Feng <[email protected]> * Add compat test Signed-off-by: Karen Feng <[email protected]> * scalafmt Signed-off-by: Karen Feng <[email protected]> * collect minimal columns Signed-off-by: Karen Feng <[email protected]> * address comments Signed-off-by: Karen Feng <[email protected]> * Test fixup Signed-off-by: Karen Feng <[email protected]> * Spark 3 needs more recent PyArrow, reduce mem consumption by removing unnecessary caching Signed-off-by: Karen Feng <[email protected]> * PyArrow 0.15.1 only with PySpark 3 Signed-off-by: Karen Feng <[email protected]> * Don't use toPandas() Signed-off-by: Karen Feng <[email protected]> * Upgrade pyarrow Signed-off-by: Karen Feng <[email protected]> * Only register once Signed-off-by: Karen Feng <[email protected]> * Minimize memory usage Signed-off-by: Karen Feng <[email protected]> * Select before head Signed-off-by: Karen Feng <[email protected]> * set up/tear down Signed-off-by: Karen Feng <[email protected]> * Try limiting pyspark memory Signed-off-by: Karen Feng <[email protected]> * No teardown Signed-off-by: Karen Feng <[email protected]> * Extend timeout Signed-off-by: Karen Feng <[email protected]> * Simplify ordering logic in levels code (#7) * WIP Signed-off-by: Karen Feng <[email protected]> * existing tests pass Signed-off-by: Karen Feng <[email protected]> * rename file Signed-off-by: Karen Feng <[email protected]> * Add compat test Signed-off-by: Karen Feng <[email protected]> * scalafmt Signed-off-by: Karen Feng <[email protected]> * collect minimal columns Signed-off-by: Karen Feng <[email protected]> * start changing for readability * use input label ordering * rename create_row_indexer * undo column sort * change reduce Signed-off-by: Henry D <[email protected]> * further simplify reduce * sorted alpha names * remove ordering * comments Signed-off-by: Henry D <[email protected]> * Set arrow env var in build Signed-off-by: Henry D <[email protected]> * faster sort * add test file * undo test data change * >= * formatting * empty Co-authored-by: Karen Feng <[email protected]> * Limit Spark memory conf in tests (#9) * yapf Signed-off-by: Karen Feng <[email protected]> * yapf transform Signed-off-by: Karen Feng <[email protected]> * Set driver memory Signed-off-by: Karen Feng <[email protected]> * Try changing spark mem Signed-off-by: Karen Feng <[email protected]> * match java tests Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * remove driver memory flag Signed-off-by: Karen Feng <[email protected]> * Improve partitioning in block_variants_and_samples transformer (#11) Signed-off-by: kianfar77 <[email protected]> * Remove unnecessary header_block grouping (#10) * cleanup Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * cleanup Signed-off-by: Karen Feng <[email protected]> * Create sample ID blocking helper functions (#12) * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * whoops Signed-off-by: Karen Feng <[email protected]> * tests Signed-off-by: Karen Feng <[email protected]> * simplify tests Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * yapf Signed-off-by: Karen Feng <[email protected]> * index map compat Signed-off-by: Karen Feng <[email protected]> * Add docs Signed-off-by: Karen Feng <[email protected]> * Add more tests Signed-off-by: Karen Feng <[email protected]> * pass args as ints Signed-off-by: Karen Feng <[email protected]> * Don't roll our own splitter Signed-off-by: Karen Feng <[email protected]> * rename sample_index to sample_blocks Signed-off-by: Karen Feng <[email protected]> * Add type-checking to WGR APIs (#14) * Add type-checking to APIs Signed-off-by: Karen Feng <[email protected]> * Check valid alphas Signed-off-by: Karen Feng <[email protected]> * check 0 sig Signed-off-by: Karen Feng <[email protected]> * Add to install_requires list Signed-off-by: Karen Feng <[email protected]> * cleanup comments Signed-off-by: Karen Feng <[email protected]> * Add covariate support (#13) * Added necessary modifications to accomodate covariates in model fitting. The initial formulation of the WGR model assumed a form y ~ Xb, however in general we would like to use a model of the form y ~ Ca + Xb, where C is some matrix of covariates that are separate from the genomic features X. This PR makes numerous changes to accomodate covariate matrix C. Adding covariates required the following breaking changes to the APIs: * indexdf is now a required argument for RidgeReducer.transform() and RidgeRegression.transform(): * RidgeReducer.transform(blockdf, labeldf, modeldf) -> RidgeReducer.transform(blockdf, labeldf, indexdf, modeldf) * RidgeRegression.transform(blockdf, labeldf, model, cvdf) -> RidgeRegression.transform(blockdf, labeldf, indexdf, model, cvdf) Additionally, the function signatures for the fit and transform methods of RidgeReducer and RidgeRegression have all been updated to accomodate an optional covariate DataFrame as the final argument. Two new tests have been added to test_ridge_regression.py to test run modes with covariates: * test_ridge_reducer_transform_with_cov * test_two_level_regression_with_cov Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Cleaned up one unnecessary Pandas import Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Small changes for clarity and consistence with the rest of the code. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Forgot one usage of coalesce Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Added a couple of comments to explain logic and replaced usages of .values with .array Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Fixed one instance of the change .values -> .array where it was made in error. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Typo in test_ridge_regression.py. Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> * Style auto-updates with yapfAll Signed-off-by: Leland Barnard ([email protected]) Signed-off-by: Leland Barnard <[email protected]> Co-authored-by: Leland Barnard <[email protected]> Co-authored-by: Karen Feng <[email protected]> * Flatten estimated phenotypes (#15) * WIP Signed-off-by: Karen Feng <[email protected]> * Clean up tests Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * Order to match labeldf Signed-off-by: Karen Feng <[email protected]> * Check we tie-break Signed-off-by: Karen Feng <[email protected]> * cleanup Signed-off-by: Karen Feng <[email protected]> * tests Signed-off-by: Karen Feng <[email protected]> * test var name Signed-off-by: Karen Feng <[email protected]> * clean up tests Signed-off-by: Karen Feng <[email protected]> * Clean up docs Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * tests Signed-off-by: Karen Feng <[email protected]> * remove accidental files Signed-off-by: Karen Feng <[email protected]> * Add fit_transform function to models (#17) Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * Rename levels (#20) * Rename levels to wgr Signed-off-by: Karen Feng <[email protected]> * rename test files Signed-off-by: Karen Feng <[email protected]> * Add license headers (#21) * headers * executable * fix template rendering * yapf * WIP Signed-off-by: Karen Feng <[email protected]> * WIP Signed-off-by: Karen Feng <[email protected]> * More work Signed-off-by: Karen Feng <[email protected]> * More cleanup Signed-off-by: Karen Feng <[email protected]> * Fix docs tests Signed-off-by: Karen Feng <[email protected]> * address comments Signed-off-by: Karen Feng <[email protected]> * fix regression fit description Signed-off-by: Karen Feng <[email protected]> * fix capitalization Signed-off-by: Karen Feng <[email protected]> * address some comments Signed-off-by: Karen Feng <[email protected]> * more cleanup Signed-off-by: Karen Feng <[email protected]> * More cleanup Signed-off-by: Karen Feng <[email protected]> * add notebook Signed-off-by: Karen Feng <[email protected]> * update notebook Signed-off-by: Karen Feng <[email protected]> Co-authored-by: Henry D <[email protected]> Co-authored-by: Kiavash Kianfar <[email protected]> Co-authored-by: Leland <[email protected]> Co-authored-by: Leland Barnard <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
What changes are proposed in this pull request?
Helper function to register SQL extensions in PySpark.
glow.register(spark)How is this patch tested?
(Details)