mirror of
https://github.com/wassname/ray.git
synced 2026-06-29 00:39:32 +08:00
Richard Liaw
15 lines
340 B
Python
15 lines
340 B
Python
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
TuneSearchCV = None
|
|
TuneGridSearchCV = None
|
|
|
|
try:
|
|
from tune_sklearn import TuneSearchCV, TuneGridSearchCV
|
|
except ImportError:
|
|
logger.info("tune_sklearn is not installed. Please run "
|
|
"`pip install tune-sklearn`.")
|
|
|
|
__all__ = ["TuneSearchCV", "TuneGridSearchCV"]
|