ml_pipeline/ml_pipeline/training/__init__.py

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()