File size: 1,135 Bytes
d1f3f31 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 | from dataclasses import dataclass, field
from typing import List, Optional
@dataclass
class NormalizedFeature:
name: str
label: str
original_value: str
@dataclass
class NormalizedOutput:
products: List[str] = field(default_factory=list)
brands: List[str] = field(default_factory=list)
use_cases: List[str] = field(default_factory=list)
budget: int = 0
urgency: Optional[str] = None
decision_stage: Optional[str] = None
price_sensitivity: Optional[str] = None
def to_xgboost_dict(self) -> dict:
result = {"budget": self.budget}
for p in self.products:
result[f"product_{p}"] = 1
for b in self.brands:
result[f"brand_{b}"] = 1
for u in self.use_cases:
result[f"use_case_{u}"] = 1
if self.urgency:
result[f"urgency_{self.urgency}"] = 1
if self.decision_stage:
result[f"decision_stage_{self.decision_stage}"] = 1
if self.price_sensitivity:
result[f"price_sensitivity_{self.price_sensitivity}"] = 1
return result
|