wehub-resource-sync
36 lines
859 B
Python
36 lines
859 B
Python
# flake8: noqa
|
|
|
|
accuracy = 42
|
|
|
|
# __keras_hyperopt_start__
|
|
from ray import tune
|
|
from ray.tune.search.hyperopt import HyperOptSearch
|
|
import keras
|
|
|
|
|
|
def objective(config): # <1>
|
|
model = keras.models.Sequential()
|
|
model.add(keras.layers.Dense(784, activation=config["activation"]))
|
|
model.add(keras.layers.Dense(10, activation="softmax"))
|
|
|
|
model.compile(loss="binary_crossentropy", optimizer="adam", metrics=["accuracy"])
|
|
# model.fit(...)
|
|
# loss, accuracy = model.evaluate(...)
|
|
return {"accuracy": accuracy}
|
|
|
|
|
|
search_space = {"activation": tune.choice(["relu", "tanh"])} # <2>
|
|
algo = HyperOptSearch()
|
|
|
|
tuner = tune.Tuner( # <3>
|
|
objective,
|
|
tune_config=tune.TuneConfig(
|
|
metric="accuracy",
|
|
mode="max",
|
|
search_alg=algo,
|
|
),
|
|
param_space=search_space,
|
|
)
|
|
results = tuner.fit()
|
|
# __keras_hyperopt_end__
|