ins-pricing 0.3.0__py3-none-any.whl → 0.3.1__py3-none-any.whl

ins_pricing/cli/Explain_entry.py

@@ -491,54 +491,56 @@ def explain_from_config(args: argparse.Namespace) -> None:
         categorical_features = cfg.get("categorical_features")
         plot_path_style = runtime_cfg["plot_path_style"]
 
-        model = ropt.BayesOptModel(
-            train_df,
-            test_df,
-            model_name,
-            cfg["target"],
-            cfg["weight"],
-            feature_list,
-            task_type=str(cfg.get("task_type", "regression")),
-            binary_resp_nme=binary_target,
-            cate_list=categorical_features,
-            prop_test=prop_test,
-            rand_seed=rand_seed,
-            epochs=int(runtime_cfg["epochs"]),
-            use_gpu=bool(cfg.get("use_gpu", True)),
-            output_dir=output_dir,
-            xgb_max_depth_max=runtime_cfg["xgb_max_depth_max"],
-            xgb_n_estimators_max=runtime_cfg["xgb_n_estimators_max"],
-            resn_weight_decay=cfg.get("resn_weight_decay"),
-            final_ensemble=bool(cfg.get("final_ensemble", False)),
-            final_ensemble_k=int(cfg.get("final_ensemble_k", 3)),
-            final_refit=bool(cfg.get("final_refit", True)),
-            optuna_storage=runtime_cfg["optuna_storage"],
-            optuna_study_prefix=runtime_cfg["optuna_study_prefix"],
-            best_params_files=runtime_cfg["best_params_files"],
-            gnn_use_approx_knn=cfg.get("gnn_use_approx_knn", True),
-            gnn_approx_knn_threshold=cfg.get("gnn_approx_knn_threshold", 50000),
-            gnn_graph_cache=cfg.get("gnn_graph_cache"),
-            gnn_max_gpu_knn_nodes=cfg.get("gnn_max_gpu_knn_nodes", 200000),
-            gnn_knn_gpu_mem_ratio=cfg.get("gnn_knn_gpu_mem_ratio", 0.9),
-            gnn_knn_gpu_mem_overhead=cfg.get("gnn_knn_gpu_mem_overhead", 2.0),
-            region_province_col=cfg.get("region_province_col"),
-            region_city_col=cfg.get("region_city_col"),
-            region_effect_alpha=cfg.get("region_effect_alpha"),
-            geo_feature_nmes=cfg.get("geo_feature_nmes"),
-            geo_token_hidden_dim=cfg.get("geo_token_hidden_dim"),
-            geo_token_layers=cfg.get("geo_token_layers"),
-            geo_token_dropout=cfg.get("geo_token_dropout"),
-            geo_token_k_neighbors=cfg.get("geo_token_k_neighbors"),
-            geo_token_learning_rate=cfg.get("geo_token_learning_rate"),
-            geo_token_epochs=cfg.get("geo_token_epochs"),
-            ft_role=str(cfg.get("ft_role", "model")),
-            ft_feature_prefix=str(cfg.get("ft_feature_prefix", "ft_emb")),
-            ft_num_numeric_tokens=cfg.get("ft_num_numeric_tokens"),
-            infer_categorical_max_unique=int(cfg.get("infer_categorical_max_unique", 50)),
-            infer_categorical_max_ratio=float(cfg.get("infer_categorical_max_ratio", 0.05)),
-            reuse_best_params=runtime_cfg["reuse_best_params"],
-            plot_path_style=plot_path_style,
-        )
+        config_fields = getattr(ropt.BayesOptConfig, "__dataclass_fields__", {})
+        allowed_config_keys = set(config_fields.keys())
+        config_payload = {k: v for k, v in cfg.items() if k in allowed_config_keys}
+        config_payload.update({
+            "model_nme": model_name,
+            "resp_nme": cfg["target"],
+            "weight_nme": cfg["weight"],
+            "factor_nmes": feature_list,
+            "task_type": str(cfg.get("task_type", "regression")),
+            "binary_resp_nme": binary_target,
+            "cate_list": categorical_features,
+            "prop_test": prop_test,
+            "rand_seed": rand_seed,
+            "epochs": int(runtime_cfg["epochs"]),
+            "use_gpu": bool(cfg.get("use_gpu", True)),
+            "output_dir": output_dir,
+            "xgb_max_depth_max": runtime_cfg["xgb_max_depth_max"],
+            "xgb_n_estimators_max": runtime_cfg["xgb_n_estimators_max"],
+            "resn_weight_decay": cfg.get("resn_weight_decay"),
+            "final_ensemble": bool(cfg.get("final_ensemble", False)),
+            "final_ensemble_k": int(cfg.get("final_ensemble_k", 3)),
+            "final_refit": bool(cfg.get("final_refit", True)),
+            "optuna_storage": runtime_cfg["optuna_storage"],
+            "optuna_study_prefix": runtime_cfg["optuna_study_prefix"],
+            "best_params_files": runtime_cfg["best_params_files"],
+            "gnn_use_approx_knn": cfg.get("gnn_use_approx_knn", True),
+            "gnn_approx_knn_threshold": cfg.get("gnn_approx_knn_threshold", 50000),
+            "gnn_graph_cache": cfg.get("gnn_graph_cache"),
+            "gnn_max_gpu_knn_nodes": cfg.get("gnn_max_gpu_knn_nodes", 200000),
+            "gnn_knn_gpu_mem_ratio": cfg.get("gnn_knn_gpu_mem_ratio", 0.9),
+            "gnn_knn_gpu_mem_overhead": cfg.get("gnn_knn_gpu_mem_overhead", 2.0),
+            "region_province_col": cfg.get("region_province_col"),
+            "region_city_col": cfg.get("region_city_col"),
+            "region_effect_alpha": cfg.get("region_effect_alpha"),
+            "geo_feature_nmes": cfg.get("geo_feature_nmes"),
+            "geo_token_hidden_dim": cfg.get("geo_token_hidden_dim"),
+            "geo_token_layers": cfg.get("geo_token_layers"),
+            "geo_token_dropout": cfg.get("geo_token_dropout"),
+            "geo_token_k_neighbors": cfg.get("geo_token_k_neighbors"),
+            "geo_token_learning_rate": cfg.get("geo_token_learning_rate"),
+            "geo_token_epochs": cfg.get("geo_token_epochs"),
+            "ft_role": str(cfg.get("ft_role", "model")),
+            "ft_feature_prefix": str(cfg.get("ft_feature_prefix", "ft_emb")),
+            "ft_num_numeric_tokens": cfg.get("ft_num_numeric_tokens"),
+            "reuse_best_params": runtime_cfg["reuse_best_params"],
+            "plot_path_style": plot_path_style or "nested",
+        })
+        config_payload = {k: v for k, v in config_payload.items() if v is not None}
+        config = ropt.BayesOptConfig(**config_payload)
+        model = ropt.BayesOptModel(train_df, test_df, config=config)
 
         output_overrides = resolve_explain_output_overrides(
             explain_cfg,

ins_pricing/cli/bayesopt_entry_runner.py

@@ -1223,76 +1223,79 @@ def train_from_config(args: argparse.Namespace) -> None:
             cfg.get("ft_feature_prefix", args.ft_feature_prefix))
         ft_num_numeric_tokens = cfg.get("ft_num_numeric_tokens")
 
-        model = ropt.BayesOptModel(
-            train_df,
-            test_df,
-            model_name,
-            cfg["target"],
-            cfg["weight"],
-            feature_list,
-            task_type=task_type,
-            binary_resp_nme=binary_target,
-            cate_list=categorical_features,
-            prop_test=val_ratio,
-            rand_seed=rand_seed,
-            epochs=epochs,
-            use_gpu=use_gpu,
-            use_resn_data_parallel=use_resn_dp,
-            use_ft_data_parallel=use_ft_dp,
-            use_resn_ddp=use_resn_ddp,
-            use_ft_ddp=use_ft_ddp,
-            use_gnn_data_parallel=use_gnn_dp,
-            use_gnn_ddp=use_gnn_ddp,
-            output_dir=output_dir,
-            xgb_max_depth_max=xgb_max_depth_max,
-            xgb_n_estimators_max=xgb_n_estimators_max,
-            resn_weight_decay=cfg.get("resn_weight_decay"),
-            final_ensemble=bool(cfg.get("final_ensemble", False)),
-            final_ensemble_k=int(cfg.get("final_ensemble_k", 3)),
-            final_refit=bool(cfg.get("final_refit", True)),
-            optuna_storage=optuna_storage,
-            optuna_study_prefix=optuna_study_prefix,
-            best_params_files=best_params_files,
-            gnn_use_approx_knn=gnn_use_ann,
-            gnn_approx_knn_threshold=gnn_threshold,
-            gnn_graph_cache=gnn_graph_cache,
-            gnn_max_gpu_knn_nodes=gnn_max_gpu_nodes,
-            gnn_knn_gpu_mem_ratio=gnn_gpu_mem_ratio,
-            gnn_knn_gpu_mem_overhead=gnn_gpu_mem_overhead,
-            region_province_col=region_province_col,
-            region_city_col=region_city_col,
-            region_effect_alpha=region_effect_alpha,
-            geo_feature_nmes=geo_feature_nmes,
-            geo_token_hidden_dim=geo_token_hidden_dim,
-            geo_token_layers=geo_token_layers,
-            geo_token_dropout=geo_token_dropout,
-            geo_token_k_neighbors=geo_token_k_neighbors,
-            geo_token_learning_rate=geo_token_learning_rate,
-            geo_token_epochs=geo_token_epochs,
-            ft_role=ft_role,
-            ft_feature_prefix=ft_feature_prefix,
-            ft_num_numeric_tokens=ft_num_numeric_tokens,
-            infer_categorical_max_unique=int(
-                cfg.get("infer_categorical_max_unique", 50)),
-            infer_categorical_max_ratio=float(
-                cfg.get("infer_categorical_max_ratio", 0.05)),
-            reuse_best_params=reuse_best_params,
-            bo_sample_limit=bo_sample_limit,
-            cache_predictions=cache_predictions,
-            prediction_cache_dir=prediction_cache_dir,
-            prediction_cache_format=prediction_cache_format,
-            cv_strategy=cv_strategy or split_strategy,
-            cv_group_col=cv_group_col or split_group_col,
-            cv_time_col=cv_time_col or split_time_col,
-            cv_time_ascending=cv_time_ascending,
-            cv_splits=cv_splits,
-            ft_oof_folds=ft_oof_folds,
-            ft_oof_strategy=ft_oof_strategy,
-            ft_oof_shuffle=ft_oof_shuffle,
-            save_preprocess=save_preprocess,
-            preprocess_artifact_path=preprocess_artifact_path,
-            plot_path_style=plot_path_style,
-        )
+        config_fields = getattr(ropt.BayesOptConfig,
+                                "__dataclass_fields__", {})
+        allowed_config_keys = set(config_fields.keys())
+        config_payload = {k: v for k,
+                          v in cfg.items() if k in allowed_config_keys}
+        config_payload.update({
+            "model_nme": model_name,
+            "resp_nme": cfg["target"],
+            "weight_nme": cfg["weight"],
+            "factor_nmes": feature_list,
+            "task_type": task_type,
+            "binary_resp_nme": binary_target,
+            "cate_list": categorical_features,
+            "prop_test": val_ratio,
+            "rand_seed": rand_seed,
+            "epochs": epochs,
+            "use_gpu": use_gpu,
+            "use_resn_data_parallel": use_resn_dp,
+            "use_ft_data_parallel": use_ft_dp,
+            "use_gnn_data_parallel": use_gnn_dp,
+            "use_resn_ddp": use_resn_ddp,
+            "use_ft_ddp": use_ft_ddp,
+            "use_gnn_ddp": use_gnn_ddp,
+            "output_dir": output_dir,
+            "xgb_max_depth_max": xgb_max_depth_max,
+            "xgb_n_estimators_max": xgb_n_estimators_max,
+            "resn_weight_decay": cfg.get("resn_weight_decay"),
+            "final_ensemble": bool(cfg.get("final_ensemble", False)),
+            "final_ensemble_k": int(cfg.get("final_ensemble_k", 3)),
+            "final_refit": bool(cfg.get("final_refit", True)),
+            "optuna_storage": optuna_storage,
+            "optuna_study_prefix": optuna_study_prefix,
+            "best_params_files": best_params_files,
+            "gnn_use_approx_knn": gnn_use_ann,
+            "gnn_approx_knn_threshold": gnn_threshold,
+            "gnn_graph_cache": gnn_graph_cache,
+            "gnn_max_gpu_knn_nodes": gnn_max_gpu_nodes,
+            "gnn_knn_gpu_mem_ratio": gnn_gpu_mem_ratio,
+            "gnn_knn_gpu_mem_overhead": gnn_gpu_mem_overhead,
+            "region_province_col": region_province_col,
+            "region_city_col": region_city_col,
+            "region_effect_alpha": region_effect_alpha,
+            "geo_feature_nmes": geo_feature_nmes,
+            "geo_token_hidden_dim": geo_token_hidden_dim,
+            "geo_token_layers": geo_token_layers,
+            "geo_token_dropout": geo_token_dropout,
+            "geo_token_k_neighbors": geo_token_k_neighbors,
+            "geo_token_learning_rate": geo_token_learning_rate,
+            "geo_token_epochs": geo_token_epochs,
+            "ft_role": ft_role,
+            "ft_feature_prefix": ft_feature_prefix,
+            "ft_num_numeric_tokens": ft_num_numeric_tokens,
+            "reuse_best_params": reuse_best_params,
+            "bo_sample_limit": bo_sample_limit,
+            "cache_predictions": cache_predictions,
+            "prediction_cache_dir": prediction_cache_dir,
+            "prediction_cache_format": prediction_cache_format,
+            "cv_strategy": cv_strategy or split_strategy,
+            "cv_group_col": cv_group_col or split_group_col,
+            "cv_time_col": cv_time_col or split_time_col,
+            "cv_time_ascending": cv_time_ascending,
+            "cv_splits": cv_splits,
+            "ft_oof_folds": ft_oof_folds,
+            "ft_oof_strategy": ft_oof_strategy,
+            "ft_oof_shuffle": ft_oof_shuffle,
+            "save_preprocess": save_preprocess,
+            "preprocess_artifact_path": preprocess_artifact_path,
+            "plot_path_style": plot_path_style or "nested",
+        })
+        config_payload = {
+            k: v for k, v in config_payload.items() if v is not None}
+        config = ropt.BayesOptConfig(**config_payload)
+        model = ropt.BayesOptModel(train_df, test_df, config=config)
 
         if plot_requested:
             plot_cfg = cfg.get("plot", {})

ins_pricing/modelling/core/bayesopt/models/model_ft_trainer.py

@@ -626,6 +626,7 @@ class FTTransformerSklearn(TorchTrainerMixin, nn.Module):
         best_state = None
         patience_counter = 0
         is_ddp_model = isinstance(self.ft, DDP)
+        use_collectives = dist.is_initialized() and is_ddp_model
 
         clip_fn = None
         if self.device.type == 'cuda':
@@ -669,7 +670,7 @@ class FTTransformerSklearn(TorchTrainerMixin, nn.Module):
                             device=X_num_b.device)
                         local_bad = 0 if bool(torch.isfinite(batch_loss)) else 1
                         global_bad = local_bad
-                        if dist.is_initialized():
+                        if use_collectives:
                             bad = torch.tensor(
                                 [local_bad],
                                 device=batch_loss.device,
@@ -774,7 +775,7 @@ class FTTransformerSklearn(TorchTrainerMixin, nn.Module):
                             total_n += float(end - start)
                     val_loss_tensor[0] = total_val / max(total_n, 1.0)
 
-                if dist.is_initialized():
+                if use_collectives:
                     dist.broadcast(val_loss_tensor, src=0)
                 val_loss_value = float(val_loss_tensor.item())
                 prune_now = False
@@ -806,7 +807,7 @@ class FTTransformerSklearn(TorchTrainerMixin, nn.Module):
                     if trial.should_prune():
                         prune_now = True
 
-                if dist.is_initialized():
+                if use_collectives:
                     flag = torch.tensor(
                         [1 if prune_now else 0],
                         device=loss_tensor_device,

ins_pricing/modelling/core/bayesopt/trainers/trainer_base.py

@@ -1094,7 +1094,7 @@ class TrainerBase:
                 split_iter = splitter
 
         losses: List[float] = []
-        for train_idx, val_idx in split_iter:
+        for fold_idx, (train_idx, val_idx) in enumerate(split_iter):
             X_train = X_all.iloc[train_idx]
             y_train = y_all.iloc[train_idx]
             X_val = X_all.iloc[val_idx]
@@ -1108,9 +1108,11 @@ class TrainerBase:
             model = model_builder(params)
             try:
                 if fit_predict_fn:
+                    # Avoid duplicate Optuna step reports across folds.
+                    trial_for_fold = trial if fold_idx == 0 else None
                     y_pred = fit_predict_fn(
                         model, X_train, y_train, w_train,
-                        X_val, y_val, w_val, trial
+                        X_val, y_val, w_val, trial_for_fold
                     )
                 else:
                     fit_kwargs = {}
@@ -1288,4 +1290,3 @@ class TrainerBase:
             predict_kwargs_train=predict_kwargs_train,
             predict_kwargs_test=predict_kwargs_test,
             predict_fn=predict_fn)
-

ins_pricing/modelling/core/bayesopt/utils/torch_trainer_mixin.py

@@ -314,7 +314,10 @@ class TorchTrainerMixin:
             f">>> DataLoader config: Batch Size={batch_size}, Accum Steps={accum_steps}, "
             f"Workers={workers}, Prefetch={prefetch_factor or 'off'}, Profile={profile}")
         sampler = None
-        if dist.is_initialized():
+        use_distributed_sampler = bool(
+            dist.is_initialized() and getattr(self, "is_ddp_enabled", False)
+        )
+        if use_distributed_sampler:
             sampler = DistributedSampler(dataset, shuffle=True)
             shuffle = False
         else:
@@ -413,6 +416,7 @@ class TorchTrainerMixin:
         val_history: List[float] = []
 
         is_ddp_model = isinstance(model, DDP)
+        use_collectives = dist.is_initialized() and is_ddp_model
 
         for epoch in range(1, getattr(self, "epochs", 1) + 1):
             epoch_start_ts = time.time()
@@ -487,7 +491,7 @@ class TorchTrainerMixin:
                             val_weighted_loss = val_result
                     val_loss_tensor[0] = float(val_weighted_loss)
 
-                if dist.is_initialized():
+                if use_collectives:
                     dist.broadcast(val_loss_tensor, src=0)
                 val_weighted_loss = float(val_loss_tensor.item())
 
@@ -502,7 +506,7 @@ class TorchTrainerMixin:
                     trial.report(val_weighted_loss, epoch)
                     prune_flag = trial.should_prune()
 
-                if dist.is_initialized():
+                if use_collectives:
                     prune_device = getattr(self, "device", torch.device("cpu"))
                     if not isinstance(prune_device, torch.device):
                         prune_device = torch.device(prune_device)

ins_pricing/setup.py

@@ -3,7 +3,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="ins_pricing",
-    version="0.3.0",
+    version="0.3.1",
     description="Reusable modelling, pricing, governance, and reporting utilities.",
     author="meishi125478",
     license="Proprietary",

{ins_pricing-0.3.0.dist-info → ins_pricing-0.3.1.dist-info}/METADATA

@@ -1,162 +1,162 @@
-Metadata-Version: 2.4
-Name: ins_pricing
-Version: 0.3.0
-Summary: Reusable modelling, pricing, governance, and reporting utilities.
-Author: meishi125478
-License: Proprietary
-Keywords: pricing,insurance,bayesopt,ml
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3 :: Only
-Classifier: Programming Language :: Python :: 3.9
-Classifier: License :: Other/Proprietary License
-Classifier: Operating System :: OS Independent
-Classifier: Intended Audience :: Developers
-Requires-Python: >=3.9
-Description-Content-Type: text/markdown
-Requires-Dist: numpy>=1.20
-Requires-Dist: pandas>=1.4
-Provides-Extra: bayesopt
-Requires-Dist: torch>=1.13; extra == "bayesopt"
-Requires-Dist: optuna>=3.0; extra == "bayesopt"
-Requires-Dist: xgboost>=1.6; extra == "bayesopt"
-Requires-Dist: scikit-learn>=1.1; extra == "bayesopt"
-Requires-Dist: statsmodels>=0.13; extra == "bayesopt"
-Requires-Dist: joblib>=1.2; extra == "bayesopt"
-Requires-Dist: matplotlib>=3.5; extra == "bayesopt"
-Provides-Extra: plotting
-Requires-Dist: matplotlib>=3.5; extra == "plotting"
-Requires-Dist: scikit-learn>=1.1; extra == "plotting"
-Provides-Extra: explain
-Requires-Dist: torch>=1.13; extra == "explain"
-Requires-Dist: shap>=0.41; extra == "explain"
-Requires-Dist: scikit-learn>=1.1; extra == "explain"
-Provides-Extra: geo
-Requires-Dist: contextily>=1.3; extra == "geo"
-Requires-Dist: matplotlib>=3.5; extra == "geo"
-Provides-Extra: gnn
-Requires-Dist: torch>=1.13; extra == "gnn"
-Requires-Dist: pynndescent>=0.5; extra == "gnn"
-Requires-Dist: torch-geometric>=2.3; extra == "gnn"
-Provides-Extra: all
-Requires-Dist: torch>=1.13; extra == "all"
-Requires-Dist: optuna>=3.0; extra == "all"
-Requires-Dist: xgboost>=1.6; extra == "all"
-Requires-Dist: scikit-learn>=1.1; extra == "all"
-Requires-Dist: statsmodels>=0.13; extra == "all"
-Requires-Dist: joblib>=1.2; extra == "all"
-Requires-Dist: matplotlib>=3.5; extra == "all"
-Requires-Dist: shap>=0.41; extra == "all"
-Requires-Dist: contextily>=1.3; extra == "all"
-Requires-Dist: pynndescent>=0.5; extra == "all"
-Requires-Dist: torch-geometric>=2.3; extra == "all"
-
-# Insurance-Pricing
-
-A reusable toolkit for insurance modeling, pricing, governance, and reporting.
-
-## Overview
-
-Insurance-Pricing (ins_pricing) is an enterprise-grade Python library designed for machine learning model training, pricing calculations, and model governance workflows in the insurance industry.
-
-### Core Modules
-
-| Module | Description |
-|--------|-------------|
-| **modelling** | ML model training (GLM, XGBoost, ResNet, FT-Transformer, GNN) and model interpretability (SHAP, permutation importance) |
-| **pricing** | Factor table construction, numeric binning, premium calibration, exposure calculation, PSI monitoring |
-| **production** | Model prediction, batch scoring, data drift detection, production metrics monitoring |
-| **governance** | Model registry, version management, approval workflows, audit logging |
-| **reporting** | Report generation (Markdown format), report scheduling |
-| **utils** | Data validation, performance profiling, device management, logging configuration |
-
-### Quick Start
-
-```python
-# Model training with Bayesian optimization
-from ins_pricing import bayesopt as ropt
-
-model = ropt.BayesOptModel(
-    train_data, test_data,
-    model_name='my_model',
-    resp_nme='target',
-    weight_nme='weight',
-    factor_nmes=feature_list,
-    cate_list=categorical_features,
-)
-model.bayesopt_xgb(max_evals=100)      # Train XGBoost
-model.bayesopt_resnet(max_evals=50)    # Train ResNet
-model.bayesopt_ft(max_evals=50)        # Train FT-Transformer
-
-# Pricing: build factor table
-from ins_pricing.pricing import build_factor_table
-factors = build_factor_table(
-    df,
-    factor_col='age_band',
-    loss_col='claim_amount',
-    exposure_col='exposure',
-)
-
-# Production: batch scoring
-from ins_pricing.production import batch_score
-scores = batch_score(model.trainers['xgb'].predict, df)
-
-# Model governance
-from ins_pricing.governance import ModelRegistry
-registry = ModelRegistry('models.json')
-registry.register(model_name, version, metrics=metrics)
-```
-
-### Project Structure
-
-```
-ins_pricing/
-├── cli/                    # Command-line entry points
-├── modelling/
-│   ├── core/bayesopt/     # ML model training core
-│   ├── explain/           # Model interpretability
-│   └── plotting/          # Model visualization
-├── pricing/               # Insurance pricing module
-├── production/            # Production deployment module
-├── governance/            # Model governance
-├── reporting/             # Report generation
-├── utils/                 # Utilities
-└── tests/                 # Test suite
-```
-
-### Installation
-
-```bash
-# Basic installation
-pip install ins_pricing
-
-# Full installation (all optional dependencies)
-pip install ins_pricing[all]
-
-# Install specific extras
-pip install ins_pricing[bayesopt]    # Model training
-pip install ins_pricing[explain]     # Model explanation
-pip install ins_pricing[plotting]    # Visualization
-pip install ins_pricing[gnn]         # Graph neural networks
-```
-
-#### Multi-platform & GPU installation notes
-
-- **PyTorch (CPU/GPU/MPS)**: Install the correct PyTorch build for your platform/GPU first (CUDA on
-  Linux/Windows, ROCm on supported AMD platforms, or MPS on Apple Silicon). Then install the
-  optional extras you need (e.g., `bayesopt`, `explain`, or `gnn`). This avoids pip pulling a
-  mismatched wheel.
-- **Torch Geometric (GNN)**: `torch-geometric` often requires platform-specific wheels (e.g.,
-  `torch-scatter`, `torch-sparse`). Follow the official PyG installation instructions for your
-  CUDA/ROCm/CPU environment, then install `ins_pricing[gnn]`.
-- **Multi-GPU**: Training code will use CUDA when available and can enable multi-GPU via
-  `torch.distributed`/`DataParallel` where supported. On Windows, CUDA DDP is not supported and will
-  fall back to single-GPU or DataParallel where possible.
-
-### Requirements
-
-- Python >= 3.9
-- Core dependencies: numpy >= 1.20, pandas >= 1.4
-
-### License
-
-Proprietary
+Metadata-Version: 2.4
+Name: ins_pricing
+Version: 0.3.1
+Summary: Reusable modelling, pricing, governance, and reporting utilities.
+Author: meishi125478
+License: Proprietary
+Keywords: pricing,insurance,bayesopt,ml
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.9
+Classifier: License :: Other/Proprietary License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Developers
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: numpy>=1.20
+Requires-Dist: pandas>=1.4
+Provides-Extra: bayesopt
+Requires-Dist: torch>=1.13; extra == "bayesopt"
+Requires-Dist: optuna>=3.0; extra == "bayesopt"
+Requires-Dist: xgboost>=1.6; extra == "bayesopt"
+Requires-Dist: scikit-learn>=1.1; extra == "bayesopt"
+Requires-Dist: statsmodels>=0.13; extra == "bayesopt"
+Requires-Dist: joblib>=1.2; extra == "bayesopt"
+Requires-Dist: matplotlib>=3.5; extra == "bayesopt"
+Provides-Extra: plotting
+Requires-Dist: matplotlib>=3.5; extra == "plotting"
+Requires-Dist: scikit-learn>=1.1; extra == "plotting"
+Provides-Extra: explain
+Requires-Dist: torch>=1.13; extra == "explain"
+Requires-Dist: shap>=0.41; extra == "explain"
+Requires-Dist: scikit-learn>=1.1; extra == "explain"
+Provides-Extra: geo
+Requires-Dist: contextily>=1.3; extra == "geo"
+Requires-Dist: matplotlib>=3.5; extra == "geo"
+Provides-Extra: gnn
+Requires-Dist: torch>=1.13; extra == "gnn"
+Requires-Dist: pynndescent>=0.5; extra == "gnn"
+Requires-Dist: torch-geometric>=2.3; extra == "gnn"
+Provides-Extra: all
+Requires-Dist: torch>=1.13; extra == "all"
+Requires-Dist: optuna>=3.0; extra == "all"
+Requires-Dist: xgboost>=1.6; extra == "all"
+Requires-Dist: scikit-learn>=1.1; extra == "all"
+Requires-Dist: statsmodels>=0.13; extra == "all"
+Requires-Dist: joblib>=1.2; extra == "all"
+Requires-Dist: matplotlib>=3.5; extra == "all"
+Requires-Dist: shap>=0.41; extra == "all"
+Requires-Dist: contextily>=1.3; extra == "all"
+Requires-Dist: pynndescent>=0.5; extra == "all"
+Requires-Dist: torch-geometric>=2.3; extra == "all"
+
+# Insurance-Pricing
+
+A reusable toolkit for insurance modeling, pricing, governance, and reporting.
+
+## Overview
+
+Insurance-Pricing (ins_pricing) is an enterprise-grade Python library designed for machine learning model training, pricing calculations, and model governance workflows in the insurance industry.
+
+### Core Modules
+
+| Module | Description |
+|--------|-------------|
+| **modelling** | ML model training (GLM, XGBoost, ResNet, FT-Transformer, GNN) and model interpretability (SHAP, permutation importance) |
+| **pricing** | Factor table construction, numeric binning, premium calibration, exposure calculation, PSI monitoring |
+| **production** | Model prediction, batch scoring, data drift detection, production metrics monitoring |
+| **governance** | Model registry, version management, approval workflows, audit logging |
+| **reporting** | Report generation (Markdown format), report scheduling |
+| **utils** | Data validation, performance profiling, device management, logging configuration |
+
+### Quick Start
+
+```python
+# Model training with Bayesian optimization
+from ins_pricing import bayesopt as ropt
+
+model = ropt.BayesOptModel(
+    train_data, test_data,
+    model_name='my_model',
+    resp_nme='target',
+    weight_nme='weight',
+    factor_nmes=feature_list,
+    cate_list=categorical_features,
+)
+model.bayesopt_xgb(max_evals=100)      # Train XGBoost
+model.bayesopt_resnet(max_evals=50)    # Train ResNet
+model.bayesopt_ft(max_evals=50)        # Train FT-Transformer
+
+# Pricing: build factor table
+from ins_pricing.pricing import build_factor_table
+factors = build_factor_table(
+    df,
+    factor_col='age_band',
+    loss_col='claim_amount',
+    exposure_col='exposure',
+)
+
+# Production: batch scoring
+from ins_pricing.production import batch_score
+scores = batch_score(model.trainers['xgb'].predict, df)
+
+# Model governance
+from ins_pricing.governance import ModelRegistry
+registry = ModelRegistry('models.json')
+registry.register(model_name, version, metrics=metrics)
+```
+
+### Project Structure
+
+```
+ins_pricing/
+├── cli/                    # Command-line entry points
+├── modelling/
+│   ├── core/bayesopt/     # ML model training core
+│   ├── explain/           # Model interpretability
+│   └── plotting/          # Model visualization
+├── pricing/               # Insurance pricing module
+├── production/            # Production deployment module
+├── governance/            # Model governance
+├── reporting/             # Report generation
+├── utils/                 # Utilities
+└── tests/                 # Test suite
+```
+
+### Installation
+
+```bash
+# Basic installation
+pip install ins_pricing
+
+# Full installation (all optional dependencies)
+pip install ins_pricing[all]
+
+# Install specific extras
+pip install ins_pricing[bayesopt]    # Model training
+pip install ins_pricing[explain]     # Model explanation
+pip install ins_pricing[plotting]    # Visualization
+pip install ins_pricing[gnn]         # Graph neural networks
+```
+
+#### Multi-platform & GPU installation notes
+
+- **PyTorch (CPU/GPU/MPS)**: Install the correct PyTorch build for your platform/GPU first (CUDA on
+  Linux/Windows, ROCm on supported AMD platforms, or MPS on Apple Silicon). Then install the
+  optional extras you need (e.g., `bayesopt`, `explain`, or `gnn`). This avoids pip pulling a
+  mismatched wheel.
+- **Torch Geometric (GNN)**: `torch-geometric` often requires platform-specific wheels (e.g.,
+  `torch-scatter`, `torch-sparse`). Follow the official PyG installation instructions for your
+  CUDA/ROCm/CPU environment, then install `ins_pricing[gnn]`.
+- **Multi-GPU**: Training code will use CUDA when available and can enable multi-GPU via
+  `torch.distributed`/`DataParallel` where supported. On Windows, CUDA DDP is not supported and will
+  fall back to single-GPU or DataParallel where possible.
+
+### Requirements
+
+- Python >= 3.9
+- Core dependencies: numpy >= 1.20, pandas >= 1.4
+
+### License
+
+Proprietary

{ins_pricing-0.3.0.dist-info → ins_pricing-0.3.1.dist-info}/RECORD

@@ -3,14 +3,14 @@ ins_pricing/README.md,sha256=W4V2xtzM6pyQzwJPvWP7cNn-We9rxM8xrxRlBVQwoY8,3399
 ins_pricing/RELEASE_NOTES_0.2.8.md,sha256=KIJzk1jbZbZPKjwnkPSDHO_2Ipv3SP3CzCNDdf07jI0,9331
 ins_pricing/__init__.py,sha256=46j1wCdLVrgrofeBwKl-3NXTxzjbTv-w3KjW-dyKGiY,2622
 ins_pricing/exceptions.py,sha256=5fZavPV4zNJ7wPC75L215KkHXX9pRrfDAYZOdSKJMGo,4778
-ins_pricing/setup.py,sha256=Rwx8jUs1m2BbK0xw8aaHvUpr6HOln42XCmHb7QE7BmU,1702
+ins_pricing/setup.py,sha256=ehvh_bSMMpl0uerZQMZXU-NKSYgV9dlDEFvGej2Hrq0,1702
 ins_pricing/cli/BayesOpt_entry.py,sha256=X3AiNQQh5ARcjVMM2vOKWPYPDIId40n_RPZA76pTGl4,558
 ins_pricing/cli/BayesOpt_incremental.py,sha256=_Klr5vvNoq_TbgwrH_T3f0a6cHmA9iVJMViiji6ahJY,35927
 ins_pricing/cli/Explain_Run.py,sha256=gEPQjqHiXyXlCTKjUzwSvbAn5_h74ABgb_sEGs-YHVE,664
-ins_pricing/cli/Explain_entry.py,sha256=PP6j4uQspmettCk21lhh9L9jurDwZfWXGxS2EBUZpMs,23191
+ins_pricing/cli/Explain_entry.py,sha256=xS23x31qRvXqy2Wvo21wbkxmPRLHy77aZbtjY-J59NA,23570
 ins_pricing/cli/Pricing_Run.py,sha256=qZribQ_ku4NK4oIvlrLJdM-jyyKtIUoCbbvo8Wh_RQ4,711
 ins_pricing/cli/__init__.py,sha256=F296f1J_tBPv33lDJQ6LaN_CPwMJTMtOuTsMof0dr2o,50
-ins_pricing/cli/bayesopt_entry_runner.py,sha256=TKqaupVDoG0pkVBrsU5_IQidJ2dx4tms52tzptMOp_0,53365
+ins_pricing/cli/bayesopt_entry_runner.py,sha256=Cl5_uw_nHxkRESUW_OLjSJaHQMfiXdbitttN7ZS-tVM,53841
 ins_pricing/cli/watchdog_run.py,sha256=ehUkN9VqsQkxc6YC_WLanU6Pu-ers-nvPEtCaJ9UsgY,6188
 ins_pricing/cli/utils/__init__.py,sha256=u3kt1B27OiuOEgw6PQN-fNs9vNiAjdPyybsRQsZkM_I,54
 ins_pricing/cli/utils/cli_common.py,sha256=CgMnN_0NQQt7Bh5RjClOydz0LzU2TBmIOsFa3KxYQOQ,8254
@@ -44,11 +44,11 @@ ins_pricing/modelling/core/bayesopt/utils.py,sha256=fTDqBHCxsOVte0QhPqnMw8vkefIv
 ins_pricing/modelling/core/bayesopt/utils_backup.py,sha256=5RKizpR3j6KwR87WqqaXPtgjQXWPW4vM75sIkx38SSM,57924
 ins_pricing/modelling/core/bayesopt/models/__init__.py,sha256=vFFCkGnO6rm50TbxR6QekKKQjq-NW4UFwog6fng8-p8,700
 ins_pricing/modelling/core/bayesopt/models/model_ft_components.py,sha256=0I0NiDf1D3cOhTRQwatsNTw9Julmxv5v3HZV8fTrvcQ,10989
-ins_pricing/modelling/core/bayesopt/models/model_ft_trainer.py,sha256=2rOKekH20_9Ccb9X-S6adW-TZcSN9Aj1Bmym8c3Sgrc,39206
+ins_pricing/modelling/core/bayesopt/models/model_ft_trainer.py,sha256=srBsI4KmijRC6s4G-XlVI9ahzjc2yquQxgdeICP1_4A,39253
 ins_pricing/modelling/core/bayesopt/models/model_gnn.py,sha256=l_oIgLeTJndfYR8lpZbqUq0MKH6wReE2z1B8n1E0P8k,28095
 ins_pricing/modelling/core/bayesopt/models/model_resn.py,sha256=hAU77GcGC1mHbOOLfZ9vC5nhUhHlaPZmXjrkH3BrjKc,17128
 ins_pricing/modelling/core/bayesopt/trainers/__init__.py,sha256=ODYKjT-v4IDxu4ohGLCXY8r1-pMME9LAaNx6pmj5_38,481
-ins_pricing/modelling/core/bayesopt/trainers/trainer_base.py,sha256=-AaGt9pIeynG5ZcCZkemXsyjUzNbAzzWN_-hoh2Yo5Q,54788
+ins_pricing/modelling/core/bayesopt/trainers/trainer_base.py,sha256=xBo05nBYnV4crY7vQbugFkttABGKOh1qdpWILW8fazs,54961
 ins_pricing/modelling/core/bayesopt/trainers/trainer_ft.py,sha256=jkafhvfEqIV_PYJ90e8kkOXVGvFpCKS0CRSKLvQ_elQ,34730
 ins_pricing/modelling/core/bayesopt/trainers/trainer_glm.py,sha256=wVU6F2Ubyu3IxP6K-epjkUTGOE8gKPCdpVxGW-JP9rM,7806
 ins_pricing/modelling/core/bayesopt/trainers/trainer_gnn.py,sha256=OChMeBELl_XWYZmZcpKq5BLY-srtTgempk4L5D0I7ys,13453
@@ -59,7 +59,7 @@ ins_pricing/modelling/core/bayesopt/utils/constants.py,sha256=0ihYxGlJ8tIElYvkhI
 ins_pricing/modelling/core/bayesopt/utils/distributed_utils.py,sha256=cu01dHyYE5EREbmtJgCHSH6z5mQIqajz8_-oWZV6zVc,5787
 ins_pricing/modelling/core/bayesopt/utils/io_utils.py,sha256=vXDlAc_taCG2joxnC6wu0jVYA76UhRbX9OT_5z_im-E,3857
 ins_pricing/modelling/core/bayesopt/utils/metrics_and_devices.py,sha256=irrsE-oDDpOd82QyW9uxfDueF8RFKkUHME4wyWr7vRI,20533
-ins_pricing/modelling/core/bayesopt/utils/torch_trainer_mixin.py,sha256=q5aEy9Lp8Vemf3iCyWi-OGsbe9roVEbKYkN3qRJlVUo,24283
+ins_pricing/modelling/core/bayesopt/utils/torch_trainer_mixin.py,sha256=_ELpu5dxucq5luO_cqe_kxT7oMIVXd63pb4EZ7fWlUQ,24465
 ins_pricing/modelling/explain/__init__.py,sha256=CPoGzGu8TTO3FOXjxoXC13VkuIDCf3YTH6L3BqJq3Ok,1171
 ins_pricing/modelling/explain/gradients.py,sha256=9TqCws_p49nFxVMcjVxe4KCZ7frezeL0uV_LCdoM5yo,11088
 ins_pricing/modelling/explain/metrics.py,sha256=K_xOY7ZrHWhbJ79RNB7eXN3VXeTe8vq68ZLH2BlZufA,5389
@@ -119,7 +119,7 @@ ins_pricing/utils/metrics.py,sha256=zhKYgXgki8RDscjP_GO2lEgzrtMIZCqOX_aLpQzdw6k,
 ins_pricing/utils/paths.py,sha256=o_tBiclFvBci4cYg9WANwKPxrMcglEdOjDP-EZgGjdQ,8749
 ins_pricing/utils/profiling.py,sha256=kmbykHLcYywlZxAf_aVU8HXID3zOvUcBoO5Q58AijhA,11132
 ins_pricing/utils/validation.py,sha256=4Tw9VUJPk0N-WO3YUqZP-xXRl1Xpubkm0vi3WzzZrv4,13348
-ins_pricing-0.3.0.dist-info/METADATA,sha256=dCrwU2NZzA2I-VyTq85Pr5ucyoQA8gV4Un8bOMCejkg,6101
-ins_pricing-0.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-ins_pricing-0.3.0.dist-info/top_level.txt,sha256=haZuNQpHKNBEPZx3NjLnHp8pV3I_J9QG8-HyJn00FA0,12
-ins_pricing-0.3.0.dist-info/RECORD,,
+ins_pricing-0.3.1.dist-info/METADATA,sha256=6zht9zTZNt7svuM5rDbLAvawAACRb24YlvswuaC9Ab0,6263
+ins_pricing-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+ins_pricing-0.3.1.dist-info/top_level.txt,sha256=haZuNQpHKNBEPZx3NjLnHp8pV3I_J9QG8-HyJn00FA0,12
+ins_pricing-0.3.1.dist-info/RECORD,,