ins-pricing 0.4.4__py3-none-any.whl → 0.5.0__py3-none-any.whl

ins_pricing/frontend/__init__.py

@@ -1,10 +1,10 @@
-"""
-Insurance Pricing Frontend Package
-Web-based interface for configuring and running insurance pricing model tasks.
-"""
-
-from .config_builder import ConfigBuilder
-from .runner import TaskRunner, TrainingRunner
-from .ft_workflow import FTWorkflowHelper
-
-__all__ = ['ConfigBuilder', 'TaskRunner', 'TrainingRunner', 'FTWorkflowHelper']
+"""
+Insurance Pricing Frontend Package
+Web-based interface for configuring and running insurance pricing model tasks.
+"""
+
+from ins_pricing.frontend.config_builder import ConfigBuilder
+from ins_pricing.frontend.runner import TaskRunner, TrainingRunner
+from ins_pricing.frontend.ft_workflow import FTWorkflowHelper
+
+__all__ = ['ConfigBuilder', 'TaskRunner', 'TrainingRunner', 'FTWorkflowHelper']

ins_pricing/frontend/config_builder.py

@@ -32,6 +32,7 @@ class ConfigBuilder:
             "save_preprocess": False,
             "preprocess_artifact_path": None,
             "bo_sample_limit": None,
+            "build_oht": True,
             "cache_predictions": False,
             "prediction_cache_dir": None,
             "prediction_cache_format": "parquet",

ins_pricing/frontend/example_workflows.py

@@ -19,7 +19,7 @@ from ins_pricing.modelling.plotting import (
     plot_oneway,
 )
 from ins_pricing.modelling.plotting.common import finalize_figure, plt
-from ins_pricing.production.predict import load_predictor_from_config
+from ins_pricing.production.inference import load_predictor_from_config
 
 
 def _parse_csv_list(value: str) -> List[str]:

ins_pricing/governance/__init__.py

@@ -1,20 +1,20 @@
-from __future__ import annotations
-
-from .approval import ApprovalAction, ApprovalRequest, ApprovalStore
-from .audit import AuditEvent, AuditLogger
-from .registry import ModelArtifact, ModelRegistry, ModelVersion
-from .release import DeploymentState, ModelRef, ReleaseManager
-
-__all__ = [
-    "ApprovalAction",
-    "ApprovalRequest",
-    "ApprovalStore",
-    "AuditEvent",
-    "AuditLogger",
-    "ModelArtifact",
-    "ModelRegistry",
-    "ModelVersion",
-    "DeploymentState",
-    "ModelRef",
-    "ReleaseManager",
-]
+from __future__ import annotations
+
+from ins_pricing.governance.approval import ApprovalAction, ApprovalRequest, ApprovalStore
+from ins_pricing.governance.audit import AuditEvent, AuditLogger
+from ins_pricing.governance.registry import ModelArtifact, ModelRegistry, ModelVersion
+from ins_pricing.governance.release import DeploymentState, ModelRef, ReleaseManager
+
+__all__ = [
+    "ApprovalAction",
+    "ApprovalRequest",
+    "ApprovalStore",
+    "AuditEvent",
+    "AuditLogger",
+    "ModelArtifact",
+    "ModelRegistry",
+    "ModelVersion",
+    "DeploymentState",
+    "ModelRef",
+    "ReleaseManager",
+]

ins_pricing/governance/release.py

@@ -1,159 +1,159 @@
-from __future__ import annotations
-
-import json
-from dataclasses import asdict, dataclass, field
-from datetime import datetime
-from pathlib import Path
-from typing import List, Optional
-
-from .audit import AuditLogger
-from .registry import ModelRegistry
-
-
-@dataclass
-class ModelRef:
-    name: str
-    version: str
-    activated_at: str
-    actor: Optional[str] = None
-    note: Optional[str] = None
-
-
-@dataclass
-class DeploymentState:
-    env: str
-    active: Optional[ModelRef] = None
-    history: List[ModelRef] = field(default_factory=list)
-    updated_at: Optional[str] = None
-
-
-class ReleaseManager:
-    """Environment release manager with rollback support."""
-
-    def __init__(
-        self,
-        state_dir: str | Path,
-        *,
-        registry: Optional[ModelRegistry] = None,
-        audit_logger: Optional[AuditLogger] = None,
-    ):
-        self.state_dir = Path(state_dir)
-        self.state_dir.mkdir(parents=True, exist_ok=True)
-        self.registry = registry
-        self.audit_logger = audit_logger
-
-    def _state_path(self, env: str) -> Path:
-        return self.state_dir / f"{env}.json"
-
-    def _load(self, env: str) -> DeploymentState:
-        path = self._state_path(env)
-        if not path.exists():
-            return DeploymentState(env=env)
-        with path.open("r", encoding="utf-8") as fh:
-            payload = json.load(fh)
-        active = payload.get("active")
-        history = payload.get("history", [])
-        return DeploymentState(
-            env=payload.get("env", env),
-            active=ModelRef(**active) if active else None,
-            history=[ModelRef(**item) for item in history],
-            updated_at=payload.get("updated_at"),
-        )
-
-    def _save(self, state: DeploymentState) -> None:
-        payload = {
-            "env": state.env,
-            "active": asdict(state.active) if state.active else None,
-            "history": [asdict(item) for item in state.history],
-            "updated_at": state.updated_at,
-        }
-        path = self._state_path(state.env)
-        with path.open("w", encoding="utf-8") as fh:
-            json.dump(payload, fh, indent=2, ensure_ascii=True)
-
-    def get_active(self, env: str) -> Optional[ModelRef]:
-        state = self._load(env)
-        return state.active
-
-    def list_history(self, env: str) -> List[ModelRef]:
-        return self._load(env).history
-
-    def deploy(
-        self,
-        env: str,
-        name: str,
-        version: str,
-        *,
-        actor: Optional[str] = None,
-        note: Optional[str] = None,
-        update_registry_status: bool = True,
-        registry_status: str = "production",
-    ) -> DeploymentState:
-        state = self._load(env)
-        if state.active and state.active.name == name and state.active.version == version:
-            return state
-
-        if state.active is not None:
-            state.history.append(state.active)
-
-        now = datetime.utcnow().isoformat()
-        state.active = ModelRef(
-            name=name,
-            version=version,
-            activated_at=now,
-            actor=actor,
-            note=note,
-        )
-        state.updated_at = now
-        self._save(state)
-
-        if self.registry and update_registry_status:
-            self.registry.promote(name, version, new_status=registry_status)
-
-        if self.audit_logger:
-            self.audit_logger.log(
-                "deploy",
-                actor or "unknown",
-                metadata={"env": env, "name": name, "version": version},
-                note=note,
-            )
-
-        return state
-
-    def rollback(
-        self,
-        env: str,
-        *,
-        actor: Optional[str] = None,
-        note: Optional[str] = None,
-        update_registry_status: bool = False,
-        registry_status: str = "production",
-    ) -> DeploymentState:
-        state = self._load(env)
-        if not state.history:
-            raise ValueError("No history available to rollback.")
-
-        previous = state.history.pop()
-        now = datetime.utcnow().isoformat()
-        state.active = ModelRef(
-            name=previous.name,
-            version=previous.version,
-            activated_at=now,
-            actor=actor or previous.actor,
-            note=note or previous.note,
-        )
-        state.updated_at = now
-        self._save(state)
-
-        if self.registry and update_registry_status:
-            self.registry.promote(previous.name, previous.version, new_status=registry_status)
-
-        if self.audit_logger:
-            self.audit_logger.log(
-                "rollback",
-                actor or "unknown",
-                metadata={"env": env, "name": previous.name, "version": previous.version},
-                note=note,
-            )
-
-        return state
+from __future__ import annotations
+
+import json
+from dataclasses import asdict, dataclass, field
+from datetime import datetime
+from pathlib import Path
+from typing import List, Optional
+
+from ins_pricing.governance.audit import AuditLogger
+from ins_pricing.governance.registry import ModelRegistry
+
+
+@dataclass
+class ModelRef:
+    name: str
+    version: str
+    activated_at: str
+    actor: Optional[str] = None
+    note: Optional[str] = None
+
+
+@dataclass
+class DeploymentState:
+    env: str
+    active: Optional[ModelRef] = None
+    history: List[ModelRef] = field(default_factory=list)
+    updated_at: Optional[str] = None
+
+
+class ReleaseManager:
+    """Environment release manager with rollback support."""
+
+    def __init__(
+        self,
+        state_dir: str | Path,
+        *,
+        registry: Optional[ModelRegistry] = None,
+        audit_logger: Optional[AuditLogger] = None,
+    ):
+        self.state_dir = Path(state_dir)
+        self.state_dir.mkdir(parents=True, exist_ok=True)
+        self.registry = registry
+        self.audit_logger = audit_logger
+
+    def _state_path(self, env: str) -> Path:
+        return self.state_dir / f"{env}.json"
+
+    def _load(self, env: str) -> DeploymentState:
+        path = self._state_path(env)
+        if not path.exists():
+            return DeploymentState(env=env)
+        with path.open("r", encoding="utf-8") as fh:
+            payload = json.load(fh)
+        active = payload.get("active")
+        history = payload.get("history", [])
+        return DeploymentState(
+            env=payload.get("env", env),
+            active=ModelRef(**active) if active else None,
+            history=[ModelRef(**item) for item in history],
+            updated_at=payload.get("updated_at"),
+        )
+
+    def _save(self, state: DeploymentState) -> None:
+        payload = {
+            "env": state.env,
+            "active": asdict(state.active) if state.active else None,
+            "history": [asdict(item) for item in state.history],
+            "updated_at": state.updated_at,
+        }
+        path = self._state_path(state.env)
+        with path.open("w", encoding="utf-8") as fh:
+            json.dump(payload, fh, indent=2, ensure_ascii=True)
+
+    def get_active(self, env: str) -> Optional[ModelRef]:
+        state = self._load(env)
+        return state.active
+
+    def list_history(self, env: str) -> List[ModelRef]:
+        return self._load(env).history
+
+    def deploy(
+        self,
+        env: str,
+        name: str,
+        version: str,
+        *,
+        actor: Optional[str] = None,
+        note: Optional[str] = None,
+        update_registry_status: bool = True,
+        registry_status: str = "production",
+    ) -> DeploymentState:
+        state = self._load(env)
+        if state.active and state.active.name == name and state.active.version == version:
+            return state
+
+        if state.active is not None:
+            state.history.append(state.active)
+
+        now = datetime.utcnow().isoformat()
+        state.active = ModelRef(
+            name=name,
+            version=version,
+            activated_at=now,
+            actor=actor,
+            note=note,
+        )
+        state.updated_at = now
+        self._save(state)
+
+        if self.registry and update_registry_status:
+            self.registry.promote(name, version, new_status=registry_status)
+
+        if self.audit_logger:
+            self.audit_logger.log(
+                "deploy",
+                actor or "unknown",
+                metadata={"env": env, "name": name, "version": version},
+                note=note,
+            )
+
+        return state
+
+    def rollback(
+        self,
+        env: str,
+        *,
+        actor: Optional[str] = None,
+        note: Optional[str] = None,
+        update_registry_status: bool = False,
+        registry_status: str = "production",
+    ) -> DeploymentState:
+        state = self._load(env)
+        if not state.history:
+            raise ValueError("No history available to rollback.")
+
+        previous = state.history.pop()
+        now = datetime.utcnow().isoformat()
+        state.active = ModelRef(
+            name=previous.name,
+            version=previous.version,
+            activated_at=now,
+            actor=actor or previous.actor,
+            note=note or previous.note,
+        )
+        state.updated_at = now
+        self._save(state)
+
+        if self.registry and update_registry_status:
+            self.registry.promote(previous.name, previous.version, new_status=registry_status)
+
+        if self.audit_logger:
+            self.audit_logger.log(
+                "rollback",
+                actor or "unknown",
+                metadata={"env": env, "name": previous.name, "version": previous.version},
+                note=note,
+            )
+
+        return state

ins_pricing/modelling/README.md

@@ -0,0 +1,67 @@
+# Modelling
+
+This directory contains reusable training tooling and frameworks centered on BayesOpt.
+
+## Key locations
+
+- `core/bayesopt/` - core training/tuning package
+- `explain/` - explainability helpers
+- `plotting/` - plotting utilities
+- `ins_pricing/cli/` - CLI entry points
+- `examples/` - example configs and notebooks (repo only)
+
+## Common usage
+
+- CLI training: `python ins_pricing/cli/BayesOpt_entry.py --config-json config_template.json`
+- Notebook API: `from ins_pricing.modelling import BayesOptModel`
+
+## Explainability
+
+- CLI: `python ins_pricing/cli/Explain_entry.py --config-json config_explain_template.json`
+- Notebook: `examples/04 Explain_Run.ipynb`
+
+## Loss functions
+
+Configure the regression/classification loss with `loss_name` in the BayesOpt config.
+
+Supported `loss_name` values:
+- `auto` (default): legacy behavior based on model name
+- `tweedie`: Tweedie deviance
+- `poisson`: Poisson deviance
+- `gamma`: Gamma deviance
+- `mse`: mean squared error
+- `mae`: mean absolute error
+
+Mapping summary:
+- Tweedie deviance -> `tweedie`
+- Poisson deviance -> `poisson`
+- Gamma deviance -> `gamma`
+- Mean squared error -> `mse`
+- Mean absolute error -> `mae`
+- Classification log loss -> `logloss` (classification only)
+- Classification BCE -> `bce` (classification only)
+
+Classification tasks:
+- `loss_name` can be `auto`, `logloss`, or `bce`.
+- Training uses `BCEWithLogits` for torch models; evaluation uses log loss.
+
+Where to set `loss_name`:
+
+```json
+{
+  "task_type": "regression",
+  "loss_name": "mse"
+}
+```
+
+Behavior notes:
+- When `loss_name` is `mse` or `mae`, tuning does not sample Tweedie power.
+- When `loss_name` is `poisson` or `gamma`, power is fixed (1.0 / 2.0).
+- When `loss_name` is `tweedie`, power is sampled as usual.
+- XGBoost objective is selected from the loss name.
+
+## Notes
+
+- Models load from `output_dir/model` by default (override with `explain.model_dir`).
+- Training outputs are written to `plot/`, `Results/`, and `model/` under `output_dir`.
+- Keep large data and secrets outside the repo; use environment variables or `.env` files.