Spaces:
Sleeping
Sleeping
File size: 621 Bytes
ff0e79e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | """Utilities module"""
from .plotting import (
plot_training_curves,
plot_confusion_matrix,
plot_feature_importance,
plot_dataset_comparison,
plot_chunked_training_progress,
generate_training_report
)
from .export import (
export_to_onnx,
export_to_torchscript,
quantize_model
)
__all__ = [
'plot_training_curves',
'plot_confusion_matrix',
'plot_feature_importance',
'plot_dataset_comparison',
'plot_chunked_training_progress',
'generate_training_report',
'export_to_onnx',
'export_to_torchscript',
'quantize_model'
]
|