chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
# 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__
|
||||
Reference in New Issue
Block a user