Introduction to Advanced Machine Learning, 第二周,基于TensorFlow的MNIST实战(my1stNN)
这是HSE系列课程第一门,Introduction to Advanced Machine Learning. 第二周第二个编程作业,难易程度:中等。 使用TensorFlow对MNIST数据集图片进行分类,是一个多类分类问题。 本篇笔记对这个任务分成三个部分。 1. 实现一个二类分类问题 2. 实现一个多类分类问题,使用softmax回归,没有隐层。 3. 实现一个多类分类问题,使用softmax回归,有隐层。 from preprocessed_mnist import load_dataset X_train, y_train, X_val, y_val, X_test, y_test = load_dataset() print(X_train.shape, y_train.shape) import matplotlib.pyplot as plt %matplotlib inline plt.imshow(X_train[ 1 ], cmap= "Greys" ); (50000, 28, 28) (50000,) import tensorflow as tf s = tf.InteractiveSession() import numpy as np 1. 二类分类问题 需要将28×28的二维图片转为一维,转换完的数组加上 ‘_flatten’ 后缀。