/ESR

Primary LanguageJupyter Notebook

ESR

ESR:Exhaustive Symbolic Regression
Authors: Deaglan J. Bartlett and Harry Desmond
Homepage:https://github.com/DeaglanBartlett/ESR
Documentation:https://esr.readthedocs.io
Pre-computed function sets:https://doi.org/10.5281/zenodo.7339113
Documentation Status

About

ESR (Exhaustive Symbolic Regression) is a symbolic regression algorithm which efficiently and systematically finds all possible equations at fixed complexity (defined to be the number of nodes in its tree representation) given a set of basis functions. This is achieved by identifying the unique equations, so that one minimises the number of equations which one would have to fit to data. These unique equations are fitted to data and the results applied to the full list of equations, so we know the maximum likelihood parameters of every equation generated. We then select the best function using the minimum description length principle. By considering all equations, this procedure is guaranteed to find the true optimum for a given complexity and basis function set.

We provide all required code and plotting routines to reproduce the results of the original ESR paper, which applies this method to a catalogue of cosmic chronometers and the Pantheon+ sample of supernovae to learn the Hubble rate as a function of redshift. The functions generated for this paper and an additional function set containing exp, sqrt and square are publicly available.

Licence and Citation

Users are required to cite the Exhaustive Symbolic Regression Paper for which the following bibtex can be used

@ARTICLE{2022arXiv2211.11461,
     author = {{Bartlett}, D.~J. and {Desmond}, H. and {Ferreira}, P.~G.},
      title = "{Exhaustive Symbolic Regression}",
    journal = {arXiv e-prints},
   keywords = {Astrophysics - Cosmology and Nongalactic Astrophysics},
       year = 2022,
      month = nov,
        eid = {arXiv:2211.11461},
      pages = {arXiv:2211.11461},
archivePrefix = {arXiv},
     eprint = {2211.11461},
primaryClass = {astro-ph.CO},
        url = {https://arxiv.org/abs/2211.11461},
}

If the user uses the pre-computed function sets then they must also cite

@dataset{bartlett_deaglan_j_2022_7339113,
author       = {Bartlett, Deaglan J. and
          Desmond, Harry and
          Ferreira, Pedro G.},
title        = {Exhaustive Symbolic Regression Function Sets},
month        = nov,
year         = 2022,
note         = {{DJB is supported by the Simons Collaboration on
           ``Learning the Universe'' and was supported by
           STFC and Oriel College, Oxford. HD is supported by
           a Royal Society University Research Fellowship
           (grant no. 211046). PGF acknowledges support from
           European Research Council Grant No: 693024 and the
           Beecroft Trust.}},
publisher    = {Zenodo},
doi          = {10.5281/zenodo.7339113},
url          = {https://doi.org/10.5281/zenodo.7339113}
}

The software is available on the MIT licence:

Copyright 2022 Deaglan J. Bartlett

Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

Contributors

Below is a list of contributors to this repository.

Deaglan Bartlett (CNRS & Sorbonne Université, Institut d’Astrophysique de Paris and Astrophysics, University of Oxford)

Harry Desmond (Institute of Cosmology & Gravitation, University of Portsmouth)

Examples

To run the Pantheon example from Paper 1, one must download the Pantheon data and place in the 'data' directory.

Documentation

The documentation for this project can be found at this link

Acknowledgements

DJB is supported by the Simons Collaboration on "Learning the Universe" and was supported by STFC and Oriel College, Oxford. HD is supported by a Royal Society University Research Fellowship (grant no. 211046).