wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-19 11:27:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Remove keras dependency (#6827)

Browse Source
This commit is contained in:
Richard Liaw
2020-01-18 23:24:42 -08:00
committed by GitHub
parent c9a1810392
commit 8a9bd18606
5 changed files with 32 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/doc_code/tf_example.py
+6 -6
View File
@@ -9,12 +9,12 @@ in the documentation.
# yapf: disable
# __tf_model_start__
from tensorflow.keras import layers
def create_keras_model():
import tensorflow as tf
model = tf.keras.Sequential()
from tensorflow import keras
from tensorflow.keras import layers
model = keras.Sequential()
# Adds a densely-connected layer with 64 units to the model:
model.add(layers.Dense(64, activation="relu", input_shape=(32, )))
# Add another:
@@ -23,9 +23,9 @@ def create_keras_model():
model.add(layers.Dense(10, activation="softmax"))
model.compile(
optimizer=tf.keras.optimizers.RMSprop(0.01),
loss=tf.keras.losses.categorical_crossentropy,
metrics=[tf.keras.metrics.categorical_accuracy])
optimizer=keras.optimizers.RMSprop(0.01),
loss=keras.losses.categorical_crossentropy,
metrics=[keras.metrics.categorical_accuracy])
return model
# __tf_model_end__
# yapf: enable