mirror of
https://github.com/microsoft/autogen.git
synced 2025-10-27 15:59:35 +00:00
ca9f9054e7
* categorical choice can be ordered or unordered * ordered -> order * move choice into utils * version comparison * packaging -> setuptools * import version * version_parse * test order for choice
33 lines
751 B
Python
33 lines
751 B
Python
from flaml.tune.sample import (
|
|
BaseSampler,
|
|
PolynomialExpansionSet,
|
|
Domain,
|
|
uniform,
|
|
quniform,
|
|
randint,
|
|
qrandint,
|
|
randn,
|
|
qrandn,
|
|
loguniform,
|
|
qloguniform,
|
|
lograndint,
|
|
qlograndint,
|
|
)
|
|
from flaml.tune import choice
|
|
|
|
|
|
def test_sampler():
|
|
print(randn().sample(size=2))
|
|
print(PolynomialExpansionSet(), BaseSampler())
|
|
print(qrandn(2, 10, 2).sample(size=2))
|
|
c = choice([1, 2])
|
|
print(c.domain_str, len(c), c.is_valid(3))
|
|
c = choice([1, 2], order=False)
|
|
print(c.domain_str, len(c), c.ordered)
|
|
i = randint(1, 10)
|
|
print(i.domain_str, i.is_valid(10))
|
|
d = Domain()
|
|
print(d.domain_str, d.is_function())
|
|
d.default_sampler_cls = BaseSampler
|
|
print(d.get_sampler())
|