Files
ml_pipeline/ml_pipeline/training/__init__.py
publicmatt bae586612e refactor files.
add a notebook module.
add config package.
2024-04-05 18:37:24 -07:00

13 lines
152 B
Python

"""
main class for building a DL pipeline.
"""
from enum import Enum, auto
class Stage(Enum):
TRAIN = auto()
DEV = auto()
TEST = auto()