Skip to content

iaml_xgboost

class IAMLxgboostConfig
dataclass #

Bases: IAMLConfig

def validate() #

Validate this config.

Source code in src/mfpbench/yahpo/benchmarks/iaml/iaml_xgboost.py 
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
@no_type_check
def validate(self) -> None:
    """Validate this config."""
    assert self.booster in ["gblinear", "gbtree", "dart"]
    assert 0.00010000000000000009 <= self.alpha <= 999.9999999999998
    assert 0.00010000000000000009 <= self._lambda <= 999.9999999999998
    assert 7 <= self.nrounds <= 2981
    assert 0.1 <= self.subsample <= 1.0

    if self.colsample_bylevel is not None:
        assert self.booster in ["dart", "gbtree"]
        assert 0.01 <= self.colsample_bylevel <= 1.0

    if self.colsample_bytree is not None:
        assert self.booster in ["dart", "gbtree"]
        assert 0.01 <= self.colsample_bytree <= 1.0

    if self.eta is not None:
        assert self.booster in ["dart", "gbtree"]
        assert 0.00010000000000000009 <= self.eta <= 1.0

    if self.gamma is not None:
        assert self.booster in ["dart", "gbtree"]
        assert 0.00010000000000000009 <= self.gamma <= 6.999999999999999

    if self.max_depth is not None:
        assert self.booster in ["dart", "gbtree"]
        assert 1 <= self.max_depth <= 15

    if self.min_child_weight is not None:
        assert self.booster in ["dart", "gbtree"]
        assert 2.718281828459045 <= self.min_child_weight <= 149.99999999999997

    if self.rate_drop is not None:
        assert self.booster in ["dart"]
        assert 0.0 <= self.rate_drop <= 1.0

    if self.skip_drop is not None:
        assert self.booster in ["dart"]
        assert 0.0 <= self.skip_drop <= 1.0