plot saving methods added

This commit is contained in:
Lorenzo Volpi 2024-04-08 17:57:25 +02:00
parent 2b5c7c7e35
commit d7bb8bb2b9
1 changed files with 60 additions and 0 deletions

View File

@ -0,0 +1,60 @@
import numpy as np
from quacc.experiments.generators import get_method_names
from quacc.experiments.report import Report
from quacc.plot.matplotlib import plot_delta, plot_diagonal
def save_plot_diagonal(
basedir, cls_name, acc_name, dataset_name="*", report: Report = None
):
methods = get_method_names()
report = (
Report.load_results(
basedir,
cls_name,
acc_name,
dataset_name=dataset_name,
method_name=methods,
)
if report is None
else report
)
_methods, _true_accs, _estim_accs = report.diagonal_plot_data()
plot_diagonal(
method_names=_methods,
true_accs=_true_accs,
estim_accs=_estim_accs,
cls_name=cls_name,
acc_name=acc_name,
dataset_name=dataset_name,
basedir=basedir,
)
def save_plot_delta(
basedir, cls_name, acc_name, dataset_name="*", stdev=False, report: Report = None
):
methods = get_method_names()
report = (
Report.load_results(
basedir,
cls_name,
acc_name,
dataset_name=dataset_name,
method_name=methods,
)
if report is None
else report
)
_methods, _prevs, _acc_errs, _stdevs = report.delta_plot_data(stdev=stdev)
plot_delta(
method_names=_methods,
prevs=_prevs,
acc_errs=_acc_errs,
cls_name=cls_name,
acc_name=acc_name,
dataset_name=dataset_name,
basedir=basedir,
stdevs=_stdevs,
)