ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-10 08:35:50 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update autokeras_imdb_example.py

Browse Source
This commit is contained in:
Jake Henning 2020-04-20 15:48:39 +03:00 committed by GitHub
parent d0e04b8975
commit 0a3db7186d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/autokeras/autokeras_imdb_example.py
View File

@ -1,6 +1,6 @@
import tensorflow as tf
import autokeras as ak import autokeras as ak
import numpy as np import numpy as np
import tensorflow as tf
from tensorflow import keras from tensorflow import keras
from trains import Task from trains import Task
@ -50,4 +50,4 @@ tensorboard_callback_test = keras.callbacks.TensorBoard(log_dir='log')
clf.fit(x_train, y_train, epochs=2, callbacks=[tensorboard_callback_train]) clf.fit(x_train, y_train, epochs=2, callbacks=[tensorboard_callback_train])
clf.fit(x_test, y_test, epochs=2, callbacks=[tensorboard_callback_test]) clf.fit(x_test, y_test, epochs=2, callbacks=[tensorboard_callback_test])
# Evaluate on the testing data. # Evaluate on the testing data.
print('Accuracy: {accuracy}'.format(accuracy=clf.evaluate(x_test, y_test))) print('Accuracy: {accuracy}'.format(accuracy=clf.evaluate(x_test, y_test)))