'''
This file is used for end to end training from scratch and then, evaluation.
'''

from gcg.components import load_data, build_model, train_model, evaluate_model
from gcg import config
from gcg.utils import logging

logging.info("Initiated train pipeline")

logging.info(f"Loading data from {config.data_path}")
X_train, X_test, y_train, y_test = load_data(config.data_path, config.image_size)

logging.info("Building model...")
model = build_model(input_shape=config.image_size, num_classes=7)

logging.info("Training the model...")
train_model(model, X_train, X_test, y_train, y_test)

logging.info("Evaluating the model on test set...")
evaluate_model(model, X_test, y_test)