Creating a TensorFlow Image Classifier in Amazon SageMaker

Libraries: If you want this lab, consider one of these libraries.
AI
Cloud

Creating a TensorFlow Image Classifier in Amazon SageMaker

TensorFlow is the biggest name in machine learning frameworks. In this lab, you will use TensorFlow to create a neural network that performs a basic image classification task: deciding which LEGO brick is in an image to help you sort your giant pile of blocks.

Get started Contact sales

Lab Info

Level

Advanced

Last updated

Sep 23, 2025

Duration

1h 0m

Challenge

Navigate to the Jupyter Notebook
Challenge

Load and Prepare the Data
Challenge

Train the TensorFlow Model
Challenge

Evaluate the Model
Challenge

Make a Batch Prediction

About the author

Pluralsight Skills

Pluralsight Skills gives leaders confidence they have the skills needed to execute technology strategy. Technology teams can benchmark expertise across roles, speed up release cycles and build reliable, secure products. By leveraging our expert content, skill assessments and one-of-a-kind analytics, keep up with the pace of change, put the right people on the right projects and boost productivity. It's the most effective path to developing tech skills at scale.

Real skill practice before real-world application

Hands-on Labs are real environments created by industry experts to help you learn. These environments help you gain knowledge and experience, practice without compromising your system, test without risk, destroy without fear, and let you learn from your mistakes. Hands-on Labs: practice your skills before delivering in the real world.

Learn by doing

Engage hands-on with the tools and technologies you’re learning. You pick the skill, we provide the credentials and environment.

Follow your guide

All labs have detailed instructions and objectives, guiding you through the learning process and ensuring you understand every step.

Turn time into mastery

On average, you retain 75% more of your learning if you take time to practice. Hands-on labs set you up for success to make those skills stick.