0.0.0+develop

flytekitplugins.optuna.optimizer

Directory

Classes

Class Description
Category
Float
Integer
Number
Optimizer
Suggestion

Methods

Method Description
optimize()
process()

Variables

Property Type Description
P ParamSpec
suggest SimpleNamespace

Methods

optimize()

def optimize(
    objective: typing.Union[typing.Callable[typing.Concatenate[optuna.trial._trial.Trial, ~P], typing.Union[typing.Awaitable[typing.Union[float, tuple[float, ...]]], float, tuple[float, ...]]], flytekit.core.python_function_task.AsyncPythonFunctionTask, NoneType],
    concurrency: int,
    n_trials: int,
    study: typing.Optional[optuna.study.study.Study],
)
Parameter Type
objective typing.Union[typing.Callable[typing.Concatenate[optuna.trial._trial.Trial, ~P], typing.Union[typing.Awaitable[typing.Union[float, tuple[float, ...]]], float, tuple[float, ...]]], flytekit.core.python_function_task.AsyncPythonFunctionTask, NoneType]
concurrency int
n_trials int
study typing.Optional[optuna.study.study.Study]

process()

def process(
    trial: optuna.trial._trial.Trial,
    inputs: dict[str, typing.Any],
    root: typing.Optional[list[str]],
) -> dict[str, typing.Any]
Parameter Type
trial optuna.trial._trial.Trial
inputs dict[str, typing.Any]
root typing.Optional[list[str]]

flytekitplugins.optuna.optimizer.Category

class Category(
    choices: list[typing.Any],
)
Parameter Type
choices list[typing.Any]

flytekitplugins.optuna.optimizer.Float

class Float(
    low: float,
    high: float,
    step: typing.Optional[float],
    log: bool,
)
Parameter Type
low float
high float
step typing.Optional[float]
log bool

flytekitplugins.optuna.optimizer.Integer

class Integer(
    low: int,
    high: int,
    step: int,
    log: bool,
)
Parameter Type
low int
high int
step int
log bool

flytekitplugins.optuna.optimizer.Number

flytekitplugins.optuna.optimizer.Optimizer

class Optimizer(
    objective: typing.Union[typing.Callable[typing.Concatenate[optuna.trial._trial.Trial, ~P], typing.Union[typing.Awaitable[typing.Union[float, tuple[float, ...]]], float, tuple[float, ...]]], flytekit.core.python_function_task.AsyncPythonFunctionTask],
    concurrency: int,
    n_trials: int,
    study: typing.Optional[optuna.study.study.Study],
    delay: int,
)
Parameter Type
objective typing.Union[typing.Callable[typing.Concatenate[optuna.trial._trial.Trial, ~P], typing.Union[typing.Awaitable[typing.Union[float, tuple[float, ...]]], float, tuple[float, ...]]], flytekit.core.python_function_task.AsyncPythonFunctionTask]
concurrency int
n_trials int
study typing.Optional[optuna.study.study.Study]
delay int

Methods

Method Description
spawn()

spawn()

def spawn(
    semaphore: asyncio.locks.Semaphore,
    inputs: dict[str, typing.Any],
)
Parameter Type
semaphore asyncio.locks.Semaphore
inputs dict[str, typing.Any]

Properties

Property Type Description
is_imperative

flytekitplugins.optuna.optimizer.Suggestion