dragon-ml-toolbox 13.8.0__py3-none-any.whl → 14.0.0__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of dragon-ml-toolbox might be problematic. Click here for more details.
- {dragon_ml_toolbox-13.8.0.dist-info → dragon_ml_toolbox-14.0.0.dist-info}/METADATA +2 -1
- {dragon_ml_toolbox-13.8.0.dist-info → dragon_ml_toolbox-14.0.0.dist-info}/RECORD +21 -14
- ml_tools/ML_datasetmaster.py +2 -185
- ml_tools/ML_evaluation.py +3 -3
- ml_tools/ML_inference.py +0 -1
- ml_tools/ML_models.py +3 -1
- ml_tools/ML_trainer.py +446 -11
- ml_tools/ML_utilities.py +50 -1
- ml_tools/ML_vision_datasetmaster.py +1315 -0
- ml_tools/ML_vision_evaluation.py +260 -0
- ml_tools/ML_vision_inference.py +428 -0
- ml_tools/ML_vision_models.py +627 -0
- ml_tools/ML_vision_transformers.py +58 -0
- ml_tools/_ML_pytorch_tabular.py +543 -0
- ml_tools/_ML_vision_recipe.py +88 -0
- ml_tools/custom_logger.py +11 -6
- ml_tools/keys.py +30 -0
- {dragon_ml_toolbox-13.8.0.dist-info → dragon_ml_toolbox-14.0.0.dist-info}/WHEEL +0 -0
- {dragon_ml_toolbox-13.8.0.dist-info → dragon_ml_toolbox-14.0.0.dist-info}/licenses/LICENSE +0 -0
- {dragon_ml_toolbox-13.8.0.dist-info → dragon_ml_toolbox-14.0.0.dist-info}/licenses/LICENSE-THIRD-PARTY.md +0 -0
- {dragon_ml_toolbox-13.8.0.dist-info → dragon_ml_toolbox-14.0.0.dist-info}/top_level.txt +0 -0
ml_tools/keys.py
CHANGED
|
@@ -36,6 +36,7 @@ class PyTorchInferenceKeys:
|
|
|
36
36
|
# For classification tasks
|
|
37
37
|
LABELS = "labels"
|
|
38
38
|
PROBABILITIES = "probabilities"
|
|
39
|
+
LABEL_NAMES = "label_names"
|
|
39
40
|
|
|
40
41
|
|
|
41
42
|
class PytorchModelArchitectureKeys:
|
|
@@ -86,6 +87,35 @@ class UtilityKeys:
|
|
|
86
87
|
TOTAL_PARAMS = "Total Parameters"
|
|
87
88
|
TRAINABLE_PARAMS = "Trainable Parameters"
|
|
88
89
|
PTH_FILE = "pth report "
|
|
90
|
+
MODEL_ARCHITECTURE_FILE = "model_architecture_summary"
|
|
91
|
+
|
|
92
|
+
|
|
93
|
+
class VisionKeys:
|
|
94
|
+
"""For vision ML metrics"""
|
|
95
|
+
SEGMENTATION_REPORT = "segmentation_report"
|
|
96
|
+
SEGMENTATION_HEATMAP = "segmentation_metrics_heatmap"
|
|
97
|
+
SEGMENTATION_CONFUSION_MATRIX = "segmentation_confusion_matrix"
|
|
98
|
+
# Object detection
|
|
99
|
+
OBJECT_DETECTION_REPORT = "object_detection_report"
|
|
100
|
+
|
|
101
|
+
|
|
102
|
+
class VisionTransformRecipeKeys:
|
|
103
|
+
"""Defines the key names for the transform recipe JSON file."""
|
|
104
|
+
TASK = "task"
|
|
105
|
+
PIPELINE = "pipeline"
|
|
106
|
+
NAME = "name"
|
|
107
|
+
KWARGS = "_kwargs"
|
|
108
|
+
PRE_TRANSFORMS = "pre_transforms"
|
|
109
|
+
RESIZE_SIZE = "resize_size"
|
|
110
|
+
CROP_SIZE = "crop_size"
|
|
111
|
+
MEAN = "mean"
|
|
112
|
+
STD = "std"
|
|
113
|
+
|
|
114
|
+
|
|
115
|
+
class ObjectDetectionKeys:
|
|
116
|
+
"""Used by the object detection dataset"""
|
|
117
|
+
BOXES = "boxes"
|
|
118
|
+
LABELS = "labels"
|
|
89
119
|
|
|
90
120
|
|
|
91
121
|
class _OneHotOtherPlaceholder:
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|