mirror of
https://github.com/wassname/ray.git
synced 2026-07-01 21:31:55 +08:00
Antoni Baum
a4f2dd2138
* Add integer quantization and loguniform support * Fix hyperopt qloguniform not being np.log'd first * Add tests, __init__ * Try to fix tests, better exceptions * Tweak docstrings * Type checks in SearchSpaceTest * Update docs * Lint, tests * Update doc/source/tune/api_docs/search_space.rst Co-authored-by: Kai Fricke <krfricke@users.noreply.github.com> Co-authored-by: Kai Fricke <krfricke@users.noreply.github.com>
36 lines
1.9 KiB
Python
36 lines
1.9 KiB
Python
from ray.tune.error import TuneError
|
|
from ray.tune.tune import run_experiments, run
|
|
from ray.tune.function_runner import with_parameters
|
|
from ray.tune.syncer import SyncConfig
|
|
from ray.tune.experiment import Experiment
|
|
from ray.tune.analysis import ExperimentAnalysis, Analysis
|
|
from ray.tune.stopper import Stopper, EarlyStopping
|
|
from ray.tune.registry import register_env, register_trainable
|
|
from ray.tune.trainable import Trainable
|
|
from ray.tune.durable_trainable import DurableTrainable
|
|
from ray.tune.callback import Callback
|
|
from ray.tune.suggest import grid_search
|
|
from ray.tune.session import (
|
|
report, get_trial_dir, get_trial_name, get_trial_id, make_checkpoint_dir,
|
|
save_checkpoint, checkpoint_dir, is_session_enabled)
|
|
from ray.tune.progress_reporter import (ProgressReporter, CLIReporter,
|
|
JupyterNotebookReporter)
|
|
from ray.tune.sample import (function, sample_from, uniform, quniform, choice,
|
|
randint, lograndint, qrandint, qlograndint, randn,
|
|
qrandn, loguniform, qloguniform)
|
|
from ray.tune.suggest import create_searcher
|
|
from ray.tune.schedulers import create_scheduler
|
|
|
|
__all__ = [
|
|
"Trainable", "DurableTrainable", "Callback", "TuneError", "grid_search",
|
|
"register_env", "register_trainable", "run", "run_experiments",
|
|
"with_parameters", "Stopper", "EarlyStopping", "Experiment", "function",
|
|
"sample_from", "track", "uniform", "quniform", "choice", "randint",
|
|
"lograndint", "qrandint", "qlograndint", "randn", "qrandn", "loguniform",
|
|
"qloguniform", "ExperimentAnalysis", "Analysis", "CLIReporter",
|
|
"JupyterNotebookReporter", "ProgressReporter", "report", "get_trial_dir",
|
|
"get_trial_name", "get_trial_id", "make_checkpoint_dir", "save_checkpoint",
|
|
"is_session_enabled", "checkpoint_dir", "SyncConfig", "create_searcher",
|
|
"create_scheduler"
|
|
]
|