MMI estimation
Dependencies
- pytorch
- joblib
- sklearn
- numpy
- scipy
- matplotlib
- tqdm
Dependencies for if-estimators
- matlab (tested with 2018b)
#find
cd <matlabroot>/extern/engines/python
python setup.py install
How to run
cd ..
# check the configuration in settings.py
python -m MMI.main
How to add new mi estimators
- go to
model/
- create a new class (see mine.py/linear_regression.py for example)
- must contain the predict function (given a N by 2 matrix, return the MI estimator (float))
- go to settings.py, add the corresponding configuration in the variable model
How to add new synthetic data
- go to
data/
- create a new class (see bimodal.py for example)
- must contain the two property: data and ground_truth (see bimodal.py for example)
- go to settings.py, add the corresponding configuration in the variable data
