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)