【ML&DL学习】1 tf_keras_classification_model

最新推荐文章于 2021-02-20 18:10:26 发布

WMXNLFD

最新推荐文章于 2021-02-20 18:10:26 发布

阅读量380

点赞数

分类专栏：机器学习之路

本文链接：https://blog.csdn.net/qq_42549254/article/details/104557212

版权

该博客介绍了如何使用tf2.0构建一个基于keras的图像分类模型，包括导入数据集、显示图片、模型创建、训练过程以及归一化处理。通过查看模型层数、参数和使用adam优化器来解决训练中的问题。

摘要由CSDN通过智能技术生成

tf2.0官方文档

https://tensorflow.google.cn/versions/r2.0/api_docs/python/tf

导入一些常用的头文件

import matplotlib as mpl
import matplotlib.pyplot as plt
%matplotlib inline
import numpy as np
import sklearn
import pandas as pd
import os
import sys
import time
import tensorflow as tf

from tensorflow import keras

print(tf.__version__)
print(sys.version_info)
for module in mpl, np, pd, sklearn, tf, keras:
    print(module.__name__, module.__version__)

在这里插入图片描述

导入数据集

# 从keras数据集 导入
fashion_mnist = keras.datasets.fashion_mnist
# 加载训练集和测试集
(x_train_all, y_train_all), (x_test, y_test) = fashion_mnist.load_data()
# 把训练集分为 验证集 和 训练集
x_valid, x_train = x_train_all[:5000], x_train_all[5000:]
y_valid, y_train = y_train_all[:5000], y_train_all[5000:]