# DATA
DATA_FOLDER = 'v2-plant-seedlings-dataset'

# MODEL FITTING
IMAGE_SIZE = 50 # 50 for testing, 150 for final model
BATCH_SIZE = 10
EPOCHS = 3 # 1 for testing, 10 for final model

# MODEL PERSISTING
MODEL_PATH = "cnn_model.h5"
PIPELINE_PATH = 'cnn_pipe.pkl'
CLASSES_PATH = 'classes.pkl'

# DATA
DATA_FOLDER = 'v2-plant-seedlings-dataset'

# MODEL FITTING
IMAGE_SIZE = 50 # 50 for testing, 150 for final model
BATCH_SIZE = 10
EPOCHS = 3 # 1 for testing, 10 for final model

# MODEL PERSISTING
MODEL_PATH = "cnn_model.h5"
PIPELINE_PATH = 'cnn_pipe.pkl'
CLASSES_PATH = 'classes.pkl'

# DATA # MODEL FITTING # 50 for testing, 150 for final model # 1 for testing, 10 for final model # MODEL PERSISTING