![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-1-1024x72.png\")
<\/figure>\n\n\n\n2. Load the dataset<\/p>\n\n\n\n
This example uses the MNIST dataset. The MNIST dataset contains 60,000 training images and 10,000 test images of handwritten digits.<\/p>\n\n\n\n
To load the MNIST dataset, use the following code:<\/p>\n\n\n\n This code reads the MNIST dataset and splits it into training and test images.<\/p>\n\n\n\n 3. Preprocess the data<\/p>\n\n\n\n MNIST images are grayscale images and are all 28×28 pixels in size. Before training the model, we need to normalize the images.<\/p>\n\n\n\n To normalize the image, use the following code:<\/p>\n\n\n\n This code normalizes the value of each pixel in the image to a range of 0 to 1.<\/p>\n\n\n\n 4. Define the model<\/p>\n\n\n\n This example uses a simple model. The model has two layers: a convolutional layer and a fully connected layer.<\/p>\n\n\n\n The convolutional layer extracts features from images. The fully connected layer classifies the image based on the features extracted by the convolutional layer.<\/p>\n\n\n\n Use the following code to define the model:<\/p>\n\n\n\n This code defines a model with four layers:<\/p>\n\n\n\n 5. Compile the model<\/p>\n\n\n\n To compile the model, you need to specify the loss function, the optimizer, and metrics.<\/p>\n\n\n\n The loss function is used to measure how well the model performs on the training data. The optimizer is used to update the weights of the model to minimize the loss function. Metrics are used to evaluate model performance on training and testing data.<\/p>\n\n\n\n Use the following code to compile the model:<\/p>\n\n\n\n This code uses “sparse_categorical_crossentropy” for the loss function, “adam” for the optimizer, and “accuracy” for the metric.<\/p>\n\n\n\n 6. Train the model<\/p>\n\n\n\n To train the model, run the code below:<\/p>\n\n\n\n This code trains the model for 10 epochs (iterations).<\/p>\n\n\n\n 7. Evaluate the model<\/p>\n\n\n\n To evaluate the model, run the code below:<\/p>\n\n\n\n This code calculates the model’s accuracy on the test data.<\/p>\n\n\n\n 8. Make predictions<\/p>\n\n\n\n To make predictions on a new image, run the code below:<\/p>\n\n\n\n This code outputs the predictions the model made on the test image.<\/p>\n\n\n\n This was the first step toward image recognition. In this blog, I introduced the basic steps for training a model for image classification using TensorFlow.<\/p>\n","protected":false},"excerpt":{"rendered":" For this engineer\u2019s blog, I would like to introduce the learning content that I am working on on a daily basis … <\/p>\n![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-2-1024x113.png\")
<\/figure>\n\n\n\n![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-3-1024x91.png\")
<\/figure>\n\n\n\n![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-4-1024x178.png\")
<\/figure>\n\n\n\n![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-5-1024x109.png\")
<\/figure>\n\n\n\n![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-6-1024x51.png\")
<\/figure>\n\n\n\n![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-7-1024x135.png\")
<\/figure>\n\n\n\n![\"\"](\"http:\/\/dandelions.co.jp\/blog\/wp\/wp-content\/uploads\/2023\/10\/image-8-1024x136.png\")
<\/figure>\n\n\n\n