Code Monkey home page Code Monkey logo

pygobn's Introduction

pyGOBN

Overview

This is a project to create a Python wrapper for the GOBNILP solver for exact Bayesian network structure learning through Integer Linear Programming. The purpose of the pyGOBN project is to facilitate easy integration of the GOBNILP solver into existing Python libraries focused on Bayesian networks.

It is currently going through some changes as I add support for actually installing pyGOBN into a user's local python distribution so that it can be used from any other module or script.

All credit for the real code behind this project goes to Dr. James Cussens and Dr. Mark Bartlett for the creation of GOBNILP, and to the creators of SCIP. Please contact me - Nick Cullen - at [email protected] with any comments or issues.

Here is the link to GOBNILP: https://www.cs.york.ac.uk/aig/sw/gobnilp/.

Here is the link to SCIP: http://scip.zib.de/#scipoptsuite.

Usage

Because of licensing issues, users will have to download the tar file for the SCIP project from the SCIP website.

The pyGOBN project comes pre-packed with the tar file of the GOBNILP source code. There is no need to download anything except this package and the SCIP tar file. However, unless you are using your own already-built GOBNILP source code, you do have to actually unpack, link, and make GOBNILP & SCIP when using pyGOBN for the first time. Luckily, all this can be done directly from Python with the single 'make()' command:

>>> from pyGOBN import *
>>> gobn = GOBN()
>>> gobn.make()

If any errors happen, they will be printed to the IPython console exactly as if you were on the real terminal. However, if you want to get a more detailed view of every step of the installation process, simply pass in the argument 'verbose=True' to the 'make()' function. Additionally, you can choose to perform each installation command seperately, although there is no real benefit to doing so:

>>> from pyGOBN import *
>>> gobn = GOBN()
>>> gobn.unpack_SCIP() # unpack SCIP tar file
>>> gobn.unpack_GOBN() # unpack GOBNILP tar file
>>> gobn.make_SCIP(verbose=True) # make SCIP with verbose output
>>> gobn.make_GOBN(verbose=True) # make GOBNILP with verbose output

Note that once you successfully 'make' the GOBNILP source code in the local pyGOBN directory system for the first time, you do NOT have to call the 'make()' command ever again. If you want to use pyGOBN again, simply import the library and create a GOBN object:

>>> from pyGOBN import *
>>> gobn = GOBN()

If you already have GOBNILP set up on your local machine, but in a different directory than the pre-packaged version, all you have to do is pass in the path to the GOBNILP directory when creating a GOBN object. The path should be of the form '.../gobnilp-1.6.1' - or whichever version you're using. Here is an example:

>>> from pyGOBN import *
>>> gobn = GOBN(GOBN_DIR='/users/nick/desktop/random_folder/gobnilp1.6.1')

Setting and altering global parameter settings for the GOBNILP solver is made simple in pyGOBN - create a settings dictionary and call the 'set_settings()' function on your GOBN object. Any unrecognized command will be passed over, but you can always directly alter the 'mysettings.txt' file in the main pyGOBN directory without calling any Python commands.

>>> gobn = GOBN()
>>> settings = {'delimiter':'whitespace', 'time':120, alpha:1000}
>>> gobn.set_settings(settings)

For more refined structure learning, you may want to add network constraints for the GOBNILP solver. Through the 'set_constraints()' method, pyGOBN supports constraints for 1) required edges, 2) disallowed edges, and 3) (conditional) independencies between random variables - all of which will be satisfied in the learned network. As with the settings file, manually changing of the 'myconstraints.txt' file is completely acceptable. See the following example:

>>> gobn = GOBN()
>>> edge_reqs = {'A':['B','C'],'B':['D']} # require that A->B, A->C, and B->D
>>> ind_reqs = [('A','D'),(('A','B'),'D','C')] # require that A _|_ D and A,B _|_ D | C
>>> nonedge_reqs = {'B':['C']} # disallow that B->C
>>> gobn.set_constraints(edge_reqs, ind_reqs, nonedge_reqs)

Finally, the purpose of GOBNILP is of course to actually learn Bayesian network structures from data. For that, pyGOBN implements the 'learn()' method, which supports learning from 1) a file path to the data, 2) a numpy array, and 3) a pandas dataframe. Here is what a short-but-complete pyGOBN session might look like:

>>> from pyGOBN import *
>>> gobn = GOBN() # Create a GOBN object
>>> settings = {'delimiter':'whitespace', 'time':120, alpha:1000}
>>> gobn.set_settings(settings) # Parameter Settings
>>> edge_reqs = {'A':['B','C'],'B':['D']} # require that A->B, A->C, and B->D
>>> gobn.set_constraints(edge_reqs=edge_reqs) # Constraints
>>> data = np.loadtxt('testdata.txt')
>>> gobn.learn(data) # Call the GOBNILP solver

Examples

Here is a real example of installing pyGOBN for the first time from the IPython shell. You can clearly see the status updates, although calling 'make()' with 'verbose=True' as an argument would print out significantly more information.

alt tag

Here is a real example of running pyGOBN from the IPython shell. In this example, the GOBNILP source code has already been installed on my local machine, so I can simply set the parameter settings - walltime equal to 10 seconds - and then learn the Bayesian network from the data file. This shows the type of output you can expect, although the level of output verbosity can easily be altered.

alt tag

Pre-Reqs

From my understanding, the only necessary requirement for setting up GOBNILP is access to the 'make' command (i.e. a compiler). This code was written for a Mac, which already has the 'make' command built-in. Since the code calls the 'tar' command to unpack the source code files, i'm assuming it's necessary to have that command as well. It's quite likely that this code will only work on Mac OS and probably Linux at the moment. If any Windows users come across any issues, please let me know at [email protected] and i'll fix it!

References

[1] Tobias Achterberg, SCIP: solving constraint integer programs, Mathematical Programming Computation, v1, n1, pgs 1-41, 2009.

[2] Mark Bartlett and James Cussens. Advances in Bayesian network learning using integer programming. In Proceedings of the 29th Conference on Uncertainty in Artificial Intelligence (UAI 2013). AUAI Press, 2013. To appear.

[3] James Cussens. Bayesian network learning with cutting planes. In Fabio G. Cozman and Avi Pfeffer, editors, Proceedings of the 27th Conference on Uncertainty in Artificial Intelligence (UAI 2011), pages 153-160. AUAI Press, 2011.

[4] James Cussens, Mark Bartlett, Elinor M. Jones, and Nuala A. Sheehan. Maximum likelihood pedigree reconstruction using

[5] Milan Studeny. How matroids occur in the context of learning Bayesian network structure. Proceedings of the 31st Conference on Uncertainty in Artificial Intelligence (UAI-15) July 2015.

[6] James Cussens, David Haws and Milan Studeny. Polyhedral aspects of score equivalence in Bayesian network structure learning. Arkiv 1503.00829, March 2015.

[7] Tommi Jaakkola, David Sontag, Amir Globerson, and Marina Meila. Learning Bayesian network structure using LP relaxations. In Proceedings of the 13th International Conference on Artificial Intelligence and Statistics (AISTATS 2010), volume 9 of Journal of Machine Learning Research: Workshop and Conference Proceedings, pages 358-365. Society for Artificial Intelligence and Statistics, 2010.

pygobn's People

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar

pygobn's Issues

Broken examples in README

Examples that don't provide a path to GOBN() are broken due to the removal of the default value for theGOBN_DIR argument.

Examples in the form gobn = GOBN() should be updated to gobn = GOBN(GOBN_DIR=...) or a sensible default set.

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.