import os

ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
DATA_DIR = os.path.join(ROOT_DIR, "data")
MODEL_DIR = os.path.join(ROOT_DIR, "models")
SRC_DIR = os.path.join(ROOT_DIR, 'src')

LOG_DIR = os.path.join(SRC_DIR, "logs")

# raw_data
RAW_DATA_PATH = os.path.join(DATA_DIR, "nyc_taxi_traffic_data.csv")

# preprocessed_data
PREPROCESSED_DATA_DIR = os.path.join(DATA_DIR, "preprocessed_data")

# features and labels path
TRAIN_FEATURES_PATH = os.path.join(PREPROCESSED_DATA_DIR, "train_features.npy")
TRAIN_LABELS_PATH = os.path.join(PREPROCESSED_DATA_DIR, "train_labels.npy")
VAL_FEATURES_PATH = os.path.join(PREPROCESSED_DATA_DIR, "val_features.npy")
VAL_LABELS_PATH = os.path.join(PREPROCESSED_DATA_DIR, "val_labels.npy")
TEST_FEATURES_PATH = os.path.join(PREPROCESSED_DATA_DIR, "test_features.npy")
TEST_LABELS_PATH = os.path.join(PREPROCESSED_DATA_DIR, "test_labels.npy")