neuro-orion-v1 / path_config.py
tgd1115's picture
Upload path_config.py
8558ff3 verified
raw
history blame contribute delete
908 Bytes
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")