Fixed minor json file that does not have all the data from a plot

leschultz · Oct 9, 2023 · 891772d · 891772d
1 parent 1d060e6
commit 891772d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -2,7 +2,7 @@
 
 # Package information
 name = 'madml'
-version = '0.9.2'  # Need to increment every time to push to PyPI
+version = '0.9.3'  # Need to increment every time to push to PyPI
 description = 'Application domain of machine learning in materials science.'
 url = 'https://github.com/leschultz/'\
       'materials_application_domain_machine_learning.git'

diff --git a/src/madml/plots.py b/src/madml/plots.py
@@ -1221,6 +1221,7 @@ def pr(score, in_domain, pos_label, save=False):
         data['precision'] = list(precision)
         data['baseline'] = baseline
         data['auc'] = auc_score
+        data['auc-baseline'] = diff
         data['auc_relative'] = auc_relative
         data.update(custom)