Source code for draugr.writers.standard_tags

from enum import Enum

__all__ = [
    "TrainingScalars",
    "TrainingCurves",
    "TrainingTables",
    "TestingScalars",
    "TestingCurves",
    "TestingTables",
    "should_plot_y_log_scale",
    "should_smooth_series",
]

from sorcery import assigned_names


[docs]class TrainingScalars(Enum): """description""" ( training_loss, validation_loss, validation_accuracy, new_best_model, ) = assigned_names()
[docs]class TrainingCurves(Enum): """description""" ( validation_precision_recall, validation_receiver_operator_characteristic, ) = assigned_names()
[docs]class TrainingTables(Enum): """description""" validation_confusion_matrix, validation_support = assigned_names()
[docs]class TestingScalars(Enum): """description""" ( test_accuracy, test_precision, test_recall, test_receiver_operator_characteristic_auc, ) = assigned_names()
[docs]class TestingCurves(Enum): """description""" test_precision_recall, test_receiver_operator_characteristic = assigned_names()
[docs]class TestingTables(Enum): """description""" test_confusion_matrix, test_support = assigned_names()
[docs]def should_plot_y_log_scale(tag: Enum) -> bool: """ :param tag: :return: """ if tag is TrainingScalars.training_loss: return True elif tag is TrainingScalars.validation_loss: return True return False
[docs]def should_smooth_series(tag: Enum) -> bool: """ :param tag: :return: """ if tag is TrainingScalars.training_loss: return True elif tag is TrainingScalars.validation_loss: return True return False
if __name__ == "__main__": print(TrainingScalars.training_loss.value)