plot saving methods added
This commit is contained in:
parent
2b5c7c7e35
commit
d7bb8bb2b9
|
@ -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,
|
||||
)
|
Loading…
Reference in New Issue