Reader small image

You're reading from  Automated Machine Learning with AutoKeras

Product typeBook
Published inMay 2021
Reading LevelBeginner
PublisherPackt
ISBN-139781800567641
Edition1st Edition
Languages
Tools
Right arrow
Author (1)
Luis Sobrecueva
Luis Sobrecueva
author image
Luis Sobrecueva

Luis Sobrecueva is a senior software engineer and ML/DL practitioner currently working at Cabify. He has been a contributor to the OpenAI project as well as one of the contributors to the AutoKeras project.
Read more about Luis Sobrecueva

Right arrow

Creating and fine-tuning a powerful image classifier

We will now use the AutoKeras ImageClassifier class to find the best classification model. Just for this little example, we set max_trials (the maximum number of different Keras models to try) to 2, and we do not set the epochs parameter so that it will use an adaptive number of epochs automatically. For real use, it is recommended to set a large number of trials. The code is shown here:

clf = ak.ImageClassifier(max_trials=2)

Let's run the training to search for the optimal classifier for the CIFAR-10 training dataset, as follows:

clf.fit(x_train, y_train)

Here is the output:

Figure 4.8 – Notebook output of image classifier training

The previous output shows that the accuracy of the training dataset is increasing.

As it has to process thousands of color images, the models that AutoKeras will generate will be more expensive to train, so this process will take hours, even using graphics...

lock icon
The rest of the page is locked
Previous PageNext Page
You have been reading a chapter from
Automated Machine Learning with AutoKeras
Published in: May 2021Publisher: PacktISBN-13: 9781800567641

Author (1)

author image
Luis Sobrecueva

Luis Sobrecueva is a senior software engineer and ML/DL practitioner currently working at Cabify. He has been a contributor to the OpenAI project as well as one of the contributors to the AutoKeras project.
Read more about Luis Sobrecueva