- tensorflow.keras.mnist.classifier.py:一个完整的mnist分类demo,其中涉及的技术点包括:
- TensorFlow dataset minist的加载
- 数据直方图打印
- 数据归一化
- label数据的 one hot vectors转换
- 数据集切分(train、test)
- CNN 模型创建
- 保存模型图片
- 图片数据增强
- 绘制训练集和验证集的loss和accuracy曲线
- 使用TensorBoard
- 对测试集做预测
- 对prediction的one-hot vector转换为数字
- 计算Precision、recall、F1等
- tensorflow.keras.save.load.model.py:讲述利用Keras api保存和加载model。
其中涉及的技术点包括:
- 保存一个模型到存储
- 加载已有模型
- 使用已有的模型做分类
- tensorflow-2.0-Alpha0-helloworld.py:基于TensorFlow2.0版本的2个完整的mnist分类demo。涉及:
- TensorFlow dataset minist的加载
- 创建自定义Model
- 对测试集做预测