https://github.com/epistasislab/scikit-mdr
A sklearn-compatible Python implementation of Multifactor Dimensionality Reduction (MDR) for feature construction.
https://github.com/epistasislab/scikit-mdr
Last synced: about 1 year ago
JSON representation
A sklearn-compatible Python implementation of Multifactor Dimensionality Reduction (MDR) for feature construction.
- Host: GitHub
- URL: https://github.com/epistasislab/scikit-mdr
- Owner: EpistasisLab
- License: mit
- Created: 2016-06-07T17:49:04.000Z (about 10 years ago)
- Default Branch: master
- Last Pushed: 2023-06-14T17:41:37.000Z (about 3 years ago)
- Last Synced: 2025-04-09T15:04:35.416Z (about 1 year ago)
- Language: Python
- Size: 138 KB
- Stars: 125
- Watchers: 12
- Forks: 26
- Open Issues: 11
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
Master status: [](https://travis-ci.org/EpistasisLab/scikit-mdr)
[](https://landscape.io/github/EpistasisLab/scikit-mdr/master)
[](https://coveralls.io/github/EpistasisLab/scikit-mdr?branch=master)
Development status: [](https://travis-ci.org/EpistasisLab/scikit-mdr)
[](https://landscape.io/github/EpistasisLab/scikit-mdr/development)
[](https://coveralls.io/github/EpistasisLab/scikit-mdr?branch=development)
Package information: 
[](https://badge.fury.io/py/scikit-MDR)
[](https://gitter.im/EpistasisLab/scikit-mdr?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge)
# MDR
A scikit-learn-compatible Python implementation of Multifactor Dimensionality Reduction (MDR) for feature construction. **This project is still under active development** and we encourage you to check back on this repository regularly for updates.
MDR is an effective feature construction algorithm that is capable of modeling higher-order interactions and capturing complex patterns in data sets.
MDR currently only works with categorical features and supports both binary classification and regression problems. We are working on expanding the algorithm to cover more problem types and provide more convenience features.
## License
Please see the [repository license](https://github.com/EpistasisLab/scikit-mdr/blob/master/LICENSE) for the licensing and usage information for the MDR package.
Generally, we have licensed the MDR package to make it as widely usable as possible.
## Installation
MDR is built on top of the following existing Python packages:
* NumPy
* SciPy
* scikit-learn
* matplotlib
All of the necessary Python packages can be installed via the [Anaconda Python distribution](https://www.continuum.io/downloads), which we strongly recommend that you use. We also strongly recommend that you use Python 3 over Python 2 if you're given the choice.
NumPy, SciPy, scikit-learn, and matplotlib can be installed in Anaconda via the command:
```
conda install numpy scipy scikit-learn matplotlib
```
Once the prerequisites are installed, you should be able to install MDR with a `pip` command:
```
pip install scikit-mdr
```
Please [file a new issue](https://github.com/EpistasisLab/scikit-mdr/issues/new) if you run into installation problems.
## Examples
MDR has been coded with a scikit-learn-like interface to be easy to use. The typical `fit`, `transform`, and `fit_transform` methods are available for every feature construction algorithm. For example, MDR can be used to construct a new feature composed from two existing features:
```python
from mdr import MDR
import pandas as pd
genetic_data = pd.read_csv('https://github.com/EpistasisLab/scikit-mdr/raw/development/data/GAMETES_Epistasis_2-Way_20atts_0.4H_EDM-1_1.tsv.gz', sep='\t', compression='gzip')
features = genetic_data.drop('class', axis=1).values
labels = genetic_data['class'].values
my_mdr = MDR()
my_mdr.fit(features, labels)
my_mdr.transform(features)
>>>array([[1],
>>> [1],
>>> [1],
>>> ...,
>>> [0],
>>> [0],
>>> [0]])
```
You can also use MDR as a classifier, and evaluate the quality of the constructed feature with the `score` function:
```python
from mdr import MDRClassifier
import pandas as pd
genetic_data = pd.read_csv('https://github.com/EpistasisLab/scikit-mdr/raw/development/data/GAMETES_Epistasis_2-Way_20atts_0.4H_EDM-1_1.tsv.gz', sep='\t', compression='gzip')
features = genetic_data.drop('class', axis=1).values
labels = genetic_data['class'].values
my_mdr = MDRClassifier()
my_mdr.fit(features, labels)
my_mdr.score(features, labels)
>>>0.998125
```
If you want to use MDR for regression problems, use `ContinuousMDR`:
```python
from mdr import ContinuousMDR
import pandas as pd
genetic_data = pd.read_csv('https://github.com/EpistasisLab/scikit-mdr/raw/development/data/GAMETES_Epistasis_2-Way_continuous_endpoint_a_20s_1600her_0.4__maf_0.2_EDM-2_01.tsv.gz', sep='\t', compression='gzip')
features = genetic_data[['M0P0', 'M0P1']].values
targets = genetic_data['Class'].values
my_cmdr = ContinuousMDR()
my_cmdr.fit(features, targets)
my_cmdr.transform(features)
>>>array([[0],
>>> [1],
>>> [1],
>>> ...,
>>> [0],
>>> [1],
>>> [1]])
```
## Contributing to MDR
We welcome you to [check the existing issues](https://github.com/EpistasisLab/scikit-mdr/issues/) for bugs or enhancements to work on. If you have an idea for an extension to the MDR package, please [file a new issue](https://github.com/EpistasisLab/scikit-mdr/issues/new) so we can discuss it.
## Having problems or have questions about MDR?
Please [check the existing open and closed issues](https://github.com/EpistasisLab/scikit-mdr/issues?utf8=%E2%9C%93&q=is%3Aissue) to see if your issue has already been attended to. If it hasn't, [file a new issue](https://github.com/EpistasisLab/scikit-mdr/issues/new) on this repository so we can review your issue.
## Citing MDR
If you use this software in a publication, please consider citing it. You can cite the repository directly with the following DOI:
[blank for now]
## Support for MDR
The MDR package was developed in the [Computational Genetics Lab](http://epistasis.org) with funding from the [NIH](http://www.nih.gov). We're incredibly grateful for their support during the development of this project.