33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import numpy as np
|
|
from skopt.space import Integer
|
|
|
|
|
|
class SKDecimal(Integer):
|
|
|
|
def __init__(self, low, high, decimals=3, prior="uniform", base=10, transform=None,
|
|
name=None, dtype=np.int64):
|
|
self.decimals = decimals
|
|
_low = int(low * pow(10, self.decimals))
|
|
_high = int(high * pow(10, self.decimals))
|
|
self.low_orig = low
|
|
self.high_orig = high
|
|
|
|
super().__init__(_low, _high, prior, base, transform, name, dtype)
|
|
|
|
def __repr__(self):
|
|
return "Decimal(low={}, high={}, decimals={}, prior='{}', transform='{}')".format(
|
|
self.low_orig, self.high_orig, self.decimals, self.prior, self.transform_)
|
|
|
|
def __contains__(self, point):
|
|
if isinstance(point, list):
|
|
point = np.array(point)
|
|
return self.low_orig <= point <= self.high_orig
|
|
|
|
def transform(self, Xt):
|
|
aa = [int(x * pow(10, self.decimals)) for x in Xt]
|
|
return super().transform(aa)
|
|
|
|
def inverse_transform(self, Xt):
|
|
res = super().inverse_transform(Xt)
|
|
return [round(x * pow(0.1, self.decimals), self.decimals) for x in res]
|