import autokeras as ak
from tensorflow.keras.datasets import mnist
(X_train_full, y_train_full), (X_test, y_test) = mnist.load_data()
Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz
11493376/11490434 [==============================] - 0s 0us/step
from sklearn.model_selection import train_test_split
_, X_train, _, y_train = train_test_split(X_train_full, y_train_full, test_size=0.1, random_state=0)
model = ak.ImageClassifier(max_trials=1)
model.fit(X_train, y_train, epochs=1)
Unsupported output type: clearOutput
Trial 1 Complete [00h 00m 13s]
val_loss: 0.1780906319618225
Best val_loss So Far: 0.1780906319618225
Total elapsed time: 00h 00m 13s
INFO:tensorflow:Oracle triggered exit
188/188 [==============================] - 15s 75ms/step - loss: 0.8854 - accuracy: 0.7175
INFO:tensorflow:Assets written to: ./image_classifier/best_model/assets
y_test_pred = model.predict(X_test)
import numpy as np
y_test_pred = np.array(y_test_pred.reshape(y_test_pred.shape[0]), int)
from sklearn.metrics import accuracy_score
accuracy_score(y_test_pred, y_test)