What is DIGEN?

Diverse and Generative ML benchmark (DIGEN) is a modern machine learning benchmark, which includes:

• 40 datasets in tabular numeric format specially designed to differentiate the performance of some of the leading Machine Learning (ML) methods, and
• a package to perform reproducible benchmarking that simplifies comparison of performance of the methods.

DIGEN provides comprehensive information on the datasets, including:

• ground truth - a mathematical formula presenting how the target was generated for each of the datasets
• the results of exploratory analysis, which includes feature correlation and histogram showing how binary endpoint was calculated.
• multiple statistics on the datasets, including the AUROC, AUPRC and F1 scores
• each dataset comes with Reveiver-Operating Characteristics (ROC) and Precision-Recall (PRC) charts for tuned ML methods,
• a boxplot with projected performance of the leading methods after hyper-parameter tuning (100 runs of each method started with different random seed)

Apart from providing a collection of datasets and tuned ML methods, DIGEN provides tools to easily tune and optimize parameters of any novel ML method, as well as visualize its performance in comparison with the leading ones. DIGEN also offers tools for reproducibility.

Dependencies

The following packages are required to use DIGEN:

pandas>=1.05
numpy>=1.19.5
optuna>=2.4.0
scikit-learn>=0.22.2
importlib_resources

Installing DIGEN

The best way to install DIGEN is using pip, e.g. as a user:

pip install -U digen

Using DIGEN

A non-peer reviewed paper is available at https://arxiv.org/pdf/2107.06475.pdf

Apart from the datasets, DIGEN provides a comprehensive toolbox for analyzing the performance of a chosen ML method. DIGEN uses Optuna, a state of the art framework for optimizing hyper-parameters

Please refer to our online documentation at https://epistasislab.github.io/digen

Citing DIGEN

If you found this resource to be helpful, please cite it the following way:

@article{orzechowski2021generative,
  title={Generative and reproducible benchmarks for comprehensive evaluation of machine learning classifiers},
  author={Orzechowski, Patryk and Moore, Jason H},
  journal={arXiv preprint arXiv:2107.06475},
  year={2021}
}

Tutorials

DIGEN Tutorial is a great place to start exploring our package. For advanced use, e.g. customization, manipulations with the charts, additional statistics on the collection, please check our Advanced Tutorial.

Included ML classifiers:

The following methods were included in our benchmark:

• Decision Tree
• Gradient Boosting
• K-Nearest Neighbors
• LightGBM
• Logistic Regression
• Random Forest
• SVC
• XGBoost