mirror of
https://github.com/wassname/optuna-dashboard.git
synced 2026-10-28 13:11:26 +08:00
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import argparse
|
|
import os
|
|
|
|
from bottle import run
|
|
from optuna.storages import BaseStorage
|
|
from optuna.storages import RDBStorage
|
|
from optuna.storages import RedisStorage
|
|
|
|
from . import __version__
|
|
from .app import create_app
|
|
|
|
|
|
AUTO_RELOAD = os.environ.get("OPTUNA_DASHBOARD_AUTO_RELOAD") == "1"
|
|
SERVER_CHOICES = ["wsgiref", "gunicorn"]
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Real-time dashboard for Optuna.")
|
|
parser.add_argument("storage", help="DB URL (e.g. sqlite:///example.db)", type=str)
|
|
parser.add_argument(
|
|
"--port", help="port number (default: %(default)s)", type=int, default=8080
|
|
)
|
|
parser.add_argument(
|
|
"--host", help="hostname (default: %(default)s)", default="127.0.0.1"
|
|
)
|
|
parser.add_argument(
|
|
"--server", help="server (default: %(default)s)", default="wsgiref", choices=SERVER_CHOICES
|
|
)
|
|
parser.add_argument("--version", "-v", action="version", version=__version__)
|
|
parser.add_argument("--quiet", "-q", help="quiet", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
storage: BaseStorage
|
|
if args.storage.startswith("redis"):
|
|
storage = RedisStorage(args.storage)
|
|
else:
|
|
storage = RDBStorage(args.storage)
|
|
|
|
app = create_app(storage)
|
|
run(app, host=args.host, port=args.port, server=args.server, quiet=args.quiet, reloader=AUTO_RELOAD)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|