Update autokeras_imdb_example.py

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

View File

@ -1,6 +1,6 @@
import tensorflow as tf
import autokeras as ak
import numpy as np
import tensorflow as tf
from tensorflow import keras
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_test, y_test, epochs=2, callbacks=[tensorboard_callback_test])
# 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)))