Skip to content

Heads

heads

Prediction heads fit on Z(X) only. Z is used unstandardized (see DECISIONS.md).

AdditiveHead 

AdditiveHead(task: str, C: float | list[float] = 1.0, alpha: float | list[float] = 1.0, random_state: int | None = None)

L2 logistic/softmax (classification) or ridge (regression) on Z(X).

Source code in src/flaggam/heads.py 
13
14
15
16
17
18
19
20
21
22
23
def __init__(
    self,
    task: str,
    C: float | list[float] = 1.0,
    alpha: float | list[float] = 1.0,
    random_state: int | None = None,
) -> None:
    self.task = task
    self.C = C
    self.alpha = alpha
    self.random_state = random_state

FlexibleHead 

FlexibleHead(estimator, task: str, random_state: int | None = None)

User-supplied tree-ensemble estimator fit on Z(X) only (no raw features).

Source code in src/flaggam/heads.py 
80
81
82
83
def __init__(self, estimator, task: str, random_state: int | None = None) -> None:
    self.estimator = estimator
    self.task = task
    self.random_state = random_state