{ "info": { "author": "Sol\u00e8ne Ulmer-Moll", "author_email": "solene.ulmer-moll@astro.up.pt", "bugtrack_url": null, "classifiers": [], "description": "## BEM : beyond the exoplanet mass-radius relation with random forest\nPredicting the radius of exoplanets based on its planetary and stellar parameters\n\n\n\n[![Build Status](https://travis-ci.org/soleneulmer/bem.svg?branch=master)](https://travis-ci.org/soleneulmer/bem)\n[![license: MIT](https://img.shields.io/badge/license-MIT-blue.svg)](https://github.com/soleneulmer/bem/LICENSE)\n[![PyPI version](https://badge.fury.io/py/bem.svg)](https://badge.fury.io/py/bem)\n[![arXiv](https://img.shields.io/badge/arXiv-1909.07392-%23B31B1B)](https://arxiv.org/abs/1909.07392)\n\n### Branca Edm\u00e9e Marques\nA portuguese scientist who worked on nuclear physics in France with Marie Curie\n\n\n### To install bem\n```bash\npip install bem\n```\nor\n```bash\ngit clone https://github.com/soleneulmer/bem.git\ncd bem\npython setup.py install\n```\n\n### A simple decision tree\n#### to predict exoplanet radius\n\n\n\n### How to run bem:\n#### 1. Load dataset and model\n```bash\n# Load exoplanet and solar system planets dataset\ndataset = bem.load_dataset()\n# Plot the dataset radius as a function of mass and equilibrium temperature\nbem.plot_dataset(dataset)\n```\n```bash\n# Build the random forest model and predict radius of the dataset\nregr, y_test_predict, _, train_test_sets = bem.random_forest_regression(dataset)\n```\n#### 2. Predict the radius of your planet\n\nmy_planet = [planetary_mass,\n semi major axis,\n eccentricity,\n stellar radius,\n stellar effective temperature,\n stellar mass]\n\nor with error bars\n\nmy_planet = [planetary_mass, error,\n semi major axis, error\n eccentricity, error,\n stellar radius, error,\n stellar effective temperature, error,\n stellar mass, error]\n```bash\n# Predict a new radius\nradius, my_pred_planet = bem.predict_radius(my_planet=np.array([[1.63,\n\t\t\t\t\t\t\t\t 0.034,\n \t\t 0.02,\n \t\t 0.337,\n \t\t 3505.0,\n \t\t 0.342]]),\n \t\t my_name=np.array(['GJ 357 b']),\n \t\t regr=regr,\n \t\t jupiter_mass=False,\n\t\t\t\t\t error_bar=False)\n# If error_bar is True\n# print('Radius: ', radius[0][0], '+-', radius[1])\n```\n\n#### 3. Compute error bars for the radius predictions\n```bash\n# Load exoplanet and solar system planets dataset with uncertainties\ndataset_errors = bem.load_dataset_errors()\n# Compute the error bars for the test set planets\nradii_test_output_error, _ = bem.computing_errorbars(regr,\n dataset_errors,\n train_test_sets)\n# Plot the test set, true radius versus RF predicted radius\nbem.plot_true_predicted(train_test_sets,\n y_test_predict,\n radii_test_output_error)\n\n```\n\n#### 4. Radial velocity dataset\n```bash\n# Load the radial velocity dataset\ndataset_rv = bem.load_dataset_RV()\n# Predict the radius of the RV dataset\nradii_RV_RF = regr.predict(dataset_rv)\n# Plot the predictions of the RV dataset\nbem.plot_dataset(dataset_rv, predicted_radii=radii_RV_RF, rv=True)\n```\n\n#### 5. Diagnostic plots\n```bash\n# Plot the learning curve\nbem.plot_learning_curve(regr, dataset)\n# Plot the validation curves\nbem.plot_validation_curves(regr, dataset, name='features')\nbem.plot_validation_curves(regr, dataset, name='tree')\nbem.plot_validation_curves(regr, dataset, name='depth')\n```\n\n#### 6. LIME explanations \nsee their [github](https://github.com/marcotcr/lime)\n```bash\n# Explain the RF predictions\n# of the exoplanets from the test set\nbem.plot_LIME_predictions(regr, dataset, train_test_sets)\n# LIME explanation for your planet\n# in this case GJ 357 b\nbem.plot_LIME_predictions(regr, dataset, train_test_sets,\n my_pred_planet=my_pred_planet,\n my_true_radius=1.166)\n```", "description_content_type": "text/markdown", "docs_url": null, "download_url": "", "downloads": { "last_day": -1, "last_month": -1, "last_week": -1 }, "home_page": "https://github.com/soleneulmer/bem", "keywords": "", "license": "MIT", "maintainer": "", "maintainer_email": "", "name": "bem", "package_url": "https://pypi.org/project/bem/", "platform": "", "project_url": "https://pypi.org/project/bem/", "project_urls": { "Homepage": "https://github.com/soleneulmer/bem" }, "release_url": "https://pypi.org/project/bem/1.0.0/", "requires_dist": null, "requires_python": "", "summary": "Random forest for exoplanets", "version": "1.0.0" }, "last_serial": 5928756, "releases": { "0.1": [ { "comment_text": "", "digests": { "md5": "d3ecb9df31926136f80a312d87536161", "sha256": "15067f22e29cb003fcdd9443b3df88adda5fa60ccc3d938a6de27a3d59602dc3" }, "downloads": -1, "filename": "bem-0.1-py3.6.egg", "has_sig": false, "md5_digest": "d3ecb9df31926136f80a312d87536161", "packagetype": "bdist_egg", "python_version": "3.6", "requires_python": null, "size": 873, "upload_time": "2019-07-24T18:13:55", "url": "https://files.pythonhosted.org/packages/a7/4b/9decd56d404a2bb8d8aa4272b6279b0ec741d76b684dcc3f5343d2af3172/bem-0.1-py3.6.egg" }, { "comment_text": "", "digests": { "md5": "2fb00fb20fcb7c084cd680c4db2db51f", "sha256": "8abaf53969b6b94c69b49aef8935172a150c8af7a34cb06bf429e6c06c49c6d5" }, "downloads": -1, "filename": "bem-0.1.tar.gz", "has_sig": false, "md5_digest": "2fb00fb20fcb7c084cd680c4db2db51f", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 717, "upload_time": "2019-07-24T18:13:58", "url": "https://files.pythonhosted.org/packages/6a/c6/2c6a08c7c41dd8a957efa063f6640de49ed0a107428469f50a5eb712b265/bem-0.1.tar.gz" } ], "0.1.1": [ { "comment_text": "", "digests": { "md5": "edc93974dd3f63d901a07970ca0f145d", "sha256": "722a2a929f4cc9d8c282cf97c92473dd72789fcfb38a4cffecc925d12aacd461" }, "downloads": -1, "filename": "bem-0.1.1-py3-none-any.whl", "has_sig": false, "md5_digest": "edc93974dd3f63d901a07970ca0f145d", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 13295, "upload_time": "2019-08-22T13:57:04", "url": "https://files.pythonhosted.org/packages/88/47/a4bd6253d65abb94093d3e18f9746db12cf5b895eb8d1c373776bb009c9c/bem-0.1.1-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "ec7a489019b0c4b951b7a2c051882b01", "sha256": "542b6039187c2c051d1741903336688cb6cb54ae9f7254642e0c054dbef294c8" }, "downloads": -1, "filename": "bem-0.1.1.tar.gz", "has_sig": false, "md5_digest": "ec7a489019b0c4b951b7a2c051882b01", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 11633, "upload_time": "2019-08-22T13:57:05", "url": "https://files.pythonhosted.org/packages/d8/51/059e2996488e3b1ac50939e315a8b41a424736a95e1d54d98651f5d5b56d/bem-0.1.1.tar.gz" } ], "0.1.11": [ { "comment_text": "", "digests": { "md5": "09d8fbeff32da0820e505e473d1f4b68", "sha256": "7550f602995fdbead038859207e195fd8e023395552e92ea73dbf507734872ae" }, "downloads": -1, "filename": "bem-0.1.11.tar.gz", "has_sig": false, "md5_digest": "09d8fbeff32da0820e505e473d1f4b68", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 743322, "upload_time": "2019-08-29T14:51:34", "url": "https://files.pythonhosted.org/packages/f2/e6/72cabdbdc4180d0b67c748cd20dee11701077a8d2e589f387247b1ffb8fa/bem-0.1.11.tar.gz" } ], "0.1.6": [ { "comment_text": "", "digests": { "md5": "3ac25c6309c8c7e5e342da92514aa738", "sha256": "107c4e420ad39ab613b6569acb06adfa009673d46d9223b61956aea0c8adb927" }, "downloads": -1, "filename": "bem-0.1.6-py3-none-any.whl", "has_sig": false, "md5_digest": "3ac25c6309c8c7e5e342da92514aa738", "packagetype": "bdist_wheel", "python_version": "py3", "requires_python": null, "size": 762155, "upload_time": "2019-08-22T16:20:57", "url": "https://files.pythonhosted.org/packages/45/15/42b6e5964bfc18a1101b104e2efa997e68306b3660ca64f120d90107fc38/bem-0.1.6-py3-none-any.whl" }, { "comment_text": "", "digests": { "md5": "092256469dc5cac8e4e7d4b991529bb7", "sha256": "ad47c4175a08fea462b53bb9af0d5889106b7c8018dc103838f64466da754745" }, "downloads": -1, "filename": "bem-0.1.6.tar.gz", "has_sig": false, "md5_digest": "092256469dc5cac8e4e7d4b991529bb7", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 738385, "upload_time": "2019-08-22T16:21:00", "url": "https://files.pythonhosted.org/packages/de/09/31fb1a10d631aa7a8857d5d6a34390eec8f8a0e5638bf93ed3766b868aa1/bem-0.1.6.tar.gz" } ], "0.1.8": [ { "comment_text": "", "digests": { "md5": "33823fc99eede0438448c271969709f6", "sha256": "ca8445bd06d974dc46a99e760d72af1b169ae484f200fa871be14a3b0949944d" }, "downloads": -1, "filename": "bem-0.1.8.tar.gz", "has_sig": false, "md5_digest": "33823fc99eede0438448c271969709f6", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 740890, "upload_time": "2019-08-23T09:58:02", "url": "https://files.pythonhosted.org/packages/71/d0/8fb0480369534f4d19d4b82943fb434989a1ace51cc59e9dd64eb81b7ef1/bem-0.1.8.tar.gz" } ], "0.1.9": [ { "comment_text": "", "digests": { "md5": "d57d18857267a8acf8360d9a806c7b4c", "sha256": "942fe5665b341356c33b65939b06aff77b793b4c71843c96d204873cb08d892c" }, "downloads": -1, "filename": "bem-0.1.9.tar.gz", "has_sig": false, "md5_digest": "d57d18857267a8acf8360d9a806c7b4c", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 742645, "upload_time": "2019-08-23T16:32:42", "url": "https://files.pythonhosted.org/packages/72/28/2ac8050d67354950ca385acd70e64f5242eccf6c72a4c1e524e89aab8121/bem-0.1.9.tar.gz" } ], "1.0.0": [ { "comment_text": "", "digests": { "md5": "c5fbecf55c002e63df4b7fe0b6f04197", "sha256": "d0abed772e2ecde8b61f0dc34440d14a173f3384de9f9b3ced793d9380ae091a" }, "downloads": -1, "filename": "bem-1.0.0.tar.gz", "has_sig": false, "md5_digest": "c5fbecf55c002e63df4b7fe0b6f04197", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 743914, "upload_time": "2019-10-04T14:47:04", "url": "https://files.pythonhosted.org/packages/b7/41/705e8f79cc54c62d37a2520f63fdc2ccc5944bd6da0c80d94b23f72e7628/bem-1.0.0.tar.gz" } ] }, "urls": [ { "comment_text": "", "digests": { "md5": "c5fbecf55c002e63df4b7fe0b6f04197", "sha256": "d0abed772e2ecde8b61f0dc34440d14a173f3384de9f9b3ced793d9380ae091a" }, "downloads": -1, "filename": "bem-1.0.0.tar.gz", "has_sig": false, "md5_digest": "c5fbecf55c002e63df4b7fe0b6f04197", "packagetype": "sdist", "python_version": "source", "requires_python": null, "size": 743914, "upload_time": "2019-10-04T14:47:04", "url": "https://files.pythonhosted.org/packages/b7/41/705e8f79cc54c62d37a2520f63fdc2ccc5944bd6da0c80d94b23f72e7628/bem-1.0.0.tar.gz" } ] }