ins-pricing 0.4.5__py3-none-any.whl → 0.5.1__py3-none-any.whl

ins_pricing/modelling/{core/bayesopt → bayesopt}/__init__.py

@@ -1,102 +1,64 @@
-"""BayesOpt subpackage (split from monolithic BayesOpt.py)."""
-
-from __future__ import annotations
-
-import torch
-
-from ins_pricing.utils.torch_compat import disable_torch_dynamo_if_requested
-
-disable_torch_dynamo_if_requested()
-
-from .config_preprocess import (
-    BayesOptConfig,
-    DatasetPreprocessor,
-    OutputManager,
-    VersionManager,
-)
-from .core import BayesOptModel
-from .models import (
-    FeatureTokenizer,
-    FTTransformerCore,
-    FTTransformerSklearn,
-    GraphNeuralNetSklearn,
-    MaskedTabularDataset,
-    ResBlock,
-    ResNetSequential,
-    ResNetSklearn,
-    ScaledTransformerEncoderLayer,
-    SimpleGraphLayer,
-    SimpleGNN,
-    TabularDataset,
-)
-from .trainers import (
-    FTTrainer,
-    GLMTrainer,
-    GNNTrainer,
-    ResNetTrainer,
-    TrainerBase,
-    XGBTrainer,
-    _xgb_cuda_available,
-)
-from .utils import (
-    EPS,
-    DistributedUtils,
-    IOUtils,
-    PlotUtils,
-    TorchTrainerMixin,
-    TrainingUtils,
-    compute_batch_size,
-    csv_to_dict,
-    ensure_parent_dir,
-    free_cuda,
-    infer_factor_and_cate_list,
-    plot_dlift_list,
-    plot_lift_list,
-    set_global_seed,
-    split_data,
-    tweedie_loss,
-)
-
-__all__ = [
-    "BayesOptConfig",
-    "DatasetPreprocessor",
-    "OutputManager",
-    "VersionManager",
-    "BayesOptModel",
-    "FeatureTokenizer",
-    "FTTransformerCore",
-    "FTTransformerSklearn",
-    "GraphNeuralNetSklearn",
-    "MaskedTabularDataset",
-    "ResBlock",
-    "ResNetSequential",
-    "ResNetSklearn",
-    "ScaledTransformerEncoderLayer",
-    "SimpleGraphLayer",
-    "SimpleGNN",
-    "TabularDataset",
-    "FTTrainer",
-    "GLMTrainer",
-    "GNNTrainer",
-    "ResNetTrainer",
-    "TrainerBase",
-    "XGBTrainer",
-    "_xgb_cuda_available",
-    "EPS",
-    "DistributedUtils",
-    "IOUtils",
-    "PlotUtils",
-    "TorchTrainerMixin",
-    "TrainingUtils",
-    "compute_batch_size",
-    "csv_to_dict",
-    "ensure_parent_dir",
-    "free_cuda",
-    "infer_factor_and_cate_list",
-    "plot_dlift_list",
-    "plot_lift_list",
-    "set_global_seed",
-    "split_data",
-    "tweedie_loss",
-    "torch",
-]
+"""BayesOpt subpackage (split from monolithic BayesOpt.py)."""
+
+from __future__ import annotations
+
+from ins_pricing.utils.torch_compat import disable_torch_dynamo_if_requested
+
+disable_torch_dynamo_if_requested()
+
+from ins_pricing.modelling.bayesopt.config_preprocess import (
+    BayesOptConfig,
+    DatasetPreprocessor,
+    OutputManager,
+    VersionManager,
+)
+from ins_pricing.modelling.bayesopt.core import BayesOptModel
+from ins_pricing.modelling.bayesopt.models import (
+    FeatureTokenizer,
+    FTTransformerCore,
+    FTTransformerSklearn,
+    GraphNeuralNetSklearn,
+    MaskedTabularDataset,
+    ResBlock,
+    ResNetSequential,
+    ResNetSklearn,
+    ScaledTransformerEncoderLayer,
+    SimpleGraphLayer,
+    SimpleGNN,
+    TabularDataset,
+)
+from ins_pricing.modelling.bayesopt.trainers import (
+    FTTrainer,
+    GLMTrainer,
+    GNNTrainer,
+    ResNetTrainer,
+    TrainerBase,
+    XGBTrainer,
+    _xgb_cuda_available,
+)
+__all__ = [
+    "BayesOptConfig",
+    "DatasetPreprocessor",
+    "OutputManager",
+    "VersionManager",
+    "BayesOptModel",
+    "FeatureTokenizer",
+    "FTTransformerCore",
+    "FTTransformerSklearn",
+    "GraphNeuralNetSklearn",
+    "MaskedTabularDataset",
+    "ResBlock",
+    "ResNetSequential",
+    "ResNetSklearn",
+    "ScaledTransformerEncoderLayer",
+    "SimpleGraphLayer",
+    "SimpleGNN",
+    "TabularDataset",
+    "FTTrainer",
+    "GLMTrainer",
+    "GNNTrainer",
+    "ResNetTrainer",
+    "TrainerBase",
+    "XGBTrainer",
+    "_xgb_cuda_available",
+]

ins_pricing/modelling/{core/bayesopt → bayesopt}/config_components.py

@@ -178,10 +178,18 @@ class XGBoostConfig:
     Attributes:
         max_depth_max: Maximum tree depth for hyperparameter tuning
         n_estimators_max: Maximum number of estimators for tuning
+        gpu_id: GPU device id for XGBoost (None = default)
+        cleanup_per_fold: Whether to cleanup GPU memory after each fold
+        cleanup_synchronize: Whether to synchronize CUDA during cleanup
+        use_dmatrix: Whether to use xgb.train with DMatrix/QuantileDMatrix
     """
 
     max_depth_max: int = 25
     n_estimators_max: int = 500
+    gpu_id: Optional[int] = None
+    cleanup_per_fold: bool = False
+    cleanup_synchronize: bool = False
+    use_dmatrix: bool = True
 
     @classmethod
     def from_flat_dict(cls, d: Dict[str, Any]) -> "XGBoostConfig":
@@ -189,6 +197,10 @@ class XGBoostConfig:
         return cls(
             max_depth_max=int(d.get("xgb_max_depth_max", 25)),
             n_estimators_max=int(d.get("xgb_n_estimators_max", 500)),
+            gpu_id=d.get("xgb_gpu_id"),
+            cleanup_per_fold=bool(d.get("xgb_cleanup_per_fold", False)),
+            cleanup_synchronize=bool(d.get("xgb_cleanup_synchronize", False)),
+            use_dmatrix=bool(d.get("xgb_use_dmatrix", True)),
         )