Skip to content

Commit

Permalink
modifed API doc for uncertainty parameters added
Browse files Browse the repository at this point in the history
  • Loading branch information
spiralulam committed Nov 28, 2023
1 parent 9d2e1ba commit 47f35b9
Showing 1 changed file with 10 additions and 12 deletions.
22 changes: 10 additions & 12 deletions entmoot/models/model_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,24 +11,22 @@ class ParamValidationError(ValueError):
class UncParams:
"""
This class contains all uncertainty parameters.
:var beta: weight for penalty/exploration part in objective function
:var bound_coeff: the predictions of the GBT model are cut off, if
their absolute value exceeds bound_coeff * variance of the y-values.
:var acq_sense: "exploration": try to find good points far away from known
training data, "exploitation": stay close to explored areas and try to
find even better points there.
:var dist_trafo: controls two different types of transformations by scaling/shifting.
"normal": shift by lower bound, scale by difference of smalles and largest value
"standard": shift by mean, scale by standard deviation
:var dist_metric: compute distance measure using the l_1, the l_2 or the squared l_2 norm.
:var cat_metric: different ways to compute the distance of categorical features
"""
#: weight for penalty/exploration part in objective function
beta: float = 1.96
#: the predictions of the GBT model are cut off, if their absolute value exceeds
# bound_coeff * variance of the y-values.
bound_coeff: float = 0.5
#: "exploration": try to find good points far away from known training data,
# "exploitation": stay close to explored areas and try to find even better points there.
acq_sense: Literal["exploration", "penalty"] = "exploration"
#: controls two different types of transformations by scaling/shifting.
# "normal": shift by lower bound, scale by difference of smalles and largest value
# "standard": shift by mean, scale by standard deviation
dist_trafo: Literal["normal", "standard"] = "normal"
#: compute distance measure using the l_1, the l_2 or the squared l_2 norm.
dist_metric: Literal["euclidean_squared", "l1", "l2"] = "euclidean_squared"
#: different ways to compute the distance of categorical features
cat_metric: Literal["overlap", "of", "goodall4"] = "overlap"

def __post_init__(self):
Expand Down

0 comments on commit 47f35b9

Please sign in to comment.