Skip to content
/ pumml Public

Positive and Unlabeled Materials Machine Learning (pumml) is a code that uses semi-supervised machine learning to classify materials from only positive and unlabeled examples.

License

Notifications You must be signed in to change notification settings

ncfrey/pumml

Repository files navigation

pumml Python 3.6 Python 3.7

pumml

Positive and Unlabeled Materials Machine Learning (pumml) is a code that uses semi-supervised positive and unlabeled (PU) machine learning to classify materials when data is incomplete and only examples of "positive" materials are available. As an example, pumml was used to predict the "synthesizability" of bulk and 2D materials from "positive" examples of synthesized materials.

How to cite pumml

If you use pumml in your research, please cite the following work:

Nathan C. Frey, Jin Wang, Gabriel Iván Vega Bellido, Babak Anasori, Yury Gogotsi, and Vivek B. Shenoy. Prediction of Synthesis of 2D Metal Carbides and Nitrides (MXenes) and Their Precursors with Positive and Unlabeled Machine Learning. ACS Nano 2019 13 (3), 3031-3041.

Please also consider citing the original works that establish the underlying methodology of pumml:

Elkan, Charles, and Keith Noto. Learning classifiers from only positive and unlabeled data. Proceeding of the 14th ACM SIGKDD international conference on Knowledge discovery and data mining. ACM, 2008.

Mordelet, F.; Vert, J.-P. A Bagging SVM to Learn from Positive and Unlabeled Examples. Pattern Recognit. Lett. 2014, 37, 201−209.

Getting pumml

The easiest way to get started with pumml is to create a virtual environment with python3.6 and then pip install pumml

You can also create a virtual environment, clone this repo and do python setup.py install in the root directory.

Using pumml

In the example_notebooks folder you will find a Jupyter notebook called basic_example.ipynb that shows the basic functionality of the package. The notebook materials_project_example.ipynb shows how to use pumml to predict the synthetic accessibility of theoretical materials in the Materials Project database. Static images of Materials Project data are available on figshare for experimenting with pumml.

About pumml

More information about using PU learning for materials synthesis prediction can be found in our publication: DOI: 10.1021/acsnano.8b08014 https://pubs.acs.org/doi/abs/10.1021/acsnano.8b08014

Helpful PU learning wrappers for scikit-learn can be found at: Alexandre Drouin, pu-learning, 2013, https://github.com/aldro61/pu-learning

In addition to our transductive bagging scheme with decision tree base classifiers, we recommend the robust ensemble of support vector machines (RESVM) method introduced by Claesen et al. RESVM is an alternative PU learning method that provides an excellent benchmark. It is implemented here: Marc Claesen, EnsembleSVM, 2014, https://github.com/claesenm/EnsembleSVM and a python wrapper is available here: Marc Claesen, resvm, 2014, https://github.com/claesenm/resvm.

License

This code is made available under the MIT License.

About

Positive and Unlabeled Materials Machine Learning (pumml) is a code that uses semi-supervised machine learning to classify materials from only positive and unlabeled examples.

Topics

Resources

License

Code of conduct

Stars

Watchers

Forks

Packages

No packages published