src/training/__init__.py

"""
Training Subpackage
Contains model training pipelines for Classification (XGBoost, Random Forest, Ensemble)
and RUL Regression (LSTM, CNN-LSTM PyTorch models).
"""

import os
import sys

sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

from classification_training import train_pipeline
from rul_regression_training import run_training_pipeline, LSTMRegressor, CNNLSTMRegressor, CMAPSSDataset

__all__ = [
    'train_pipeline',
    'run_training_pipeline',
    'LSTMRegressor',
    'CNNLSTMRegressor',
    'CMAPSSDataset'
]