Spaces:
Running
Running
| import os | |
| # Define base paths | |
| BASE_DIR = os.path.dirname(os.path.abspath(__file__)) | |
| DATA_RAW_DIR = os.path.join(BASE_DIR, 'data', 'raw') | |
| DATA_PROCESSED_DIR = os.path.join(BASE_DIR, 'data', 'processed') | |
| MODELS_DIR = os.path.join(BASE_DIR, 'models') | |
| OUTPUTS_DIR = os.path.join(BASE_DIR, 'outputs') | |
| def get_raw_data_path(filename='ai4i2020.csv'): | |
| return os.path.join(DATA_RAW_DIR, filename) | |
| def get_processed_data_path(filename='features.csv'): | |
| return os.path.join(DATA_PROCESSED_DIR, filename) | |
| def get_model_path(filename): | |
| return os.path.join(MODELS_DIR, filename) | |
| def get_output_path(filename): | |
| return os.path.join(OUTPUTS_DIR, filename) | |
| # Ensure directories exist | |
| for directory in [DATA_RAW_DIR, DATA_PROCESSED_DIR, MODELS_DIR, OUTPUTS_DIR]: | |
| if not os.path.exists(directory): | |
| os.makedirs(directory) | |