TensorFlow (Google) framework học máy mã nguồn mở. Cài đặt: pip install tensorflow.
Fashion MNIST: >70.000 ảnh grayscale trang phục 10 loại, 28x28 pixel.
Chạy trên Google Colab (Python trên trình duyệt, GPU miễn phí).
Ví dụ
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
print("Phiên bản TensorFlow:", tf.__version__)
print("Đang tải dataset...")
fashion_mnist = tf.keras.datasets.fashion_mnist
print("Phân chia train/test...")
(train_images, train_labels), (test_images, test_labels) = fashion_mnist.load_data()
class_names = ['T-shirt/top', 'Trouser', 'Pullover', 'Dress', 'Coat',
'Sandal', 'Shirt', 'Sneaker', 'Bag', 'Ankle boot']
print("Kích thước dữ liệu train:", train_images.shape)
print("Số mẫu train:", len(train_labels))
print("Nhãn train:", train_labels)
print("Kích thước dữ liệu test:", test_images.shape)
print("Số mẫu test:", len(test_labels))
Nguồn mã: TensorFlow Classification Tutorial
Đầu ra
Phiên bản TensorFlow: 2.4.0 Đang tải dataset... Phân chia train/test... Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/train-labels-idx1-ubyte.gz ... Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/train-images-idx3-ubyte.gz ... Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/t10k-labels-idx1-ubyte.gz ... Downloading data from https://storage.googleapis.com/tensorflow/tf-keras-datasets/t10k-images-idx3-ubyte.gz ... Kích thước dữ liệu train: (60000, 28, 28) Số mẫu train: 60000 Nhãn train: [9 0 0 ... 3 0 5] Kích thước dữ liệu test: (10000, 28, 28) Số mẫu test: 10000
Giải thích
- Import package: tensorflow, numpy, matplotlib.
- Kiểm tra phiên bản TensorFlow.
- Tải Fashion MNIST trực tiếp từ
tf.keras.datasets. - Tách train/test: 60k train, 10k test (tổng 70k).
- Bài toán phân loại: mỗi ảnh gán 1 nhãn (10 loại trang phục).
- Hiển thị shape, số lượng mẫu, nhãn của train/test.