mirror of
https://github.com/clearml/clearml
synced 2025-01-31 09:07:00 +00:00
Update autokeras_imdb_example.py
This commit is contained in:
parent
d0e04b8975
commit
0a3db7186d
@ -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)))
|
||||
|
Loading…
Reference in New Issue
Block a user