TensorFlow-keras介绍(二)

目录

一.tf.keras相关操作

1.获取相关现有数据集

2.构建模型

3.训练与评估

二.案例:实现多层神经网络进行时装分类

1.读取数据

2.建立模型

3.设置模型参数,开始训练,评估模型

4.保存和加载模型

5.使用保存的模型进行预测

6.添加tensorboard

四.完整代码

都看到这里了,点个赞把!!!


一.tf.keras相关操作

1.获取相关现有数据集

保证在keras.datasets中,

.mnist:手写数字。

.fashion_mnist:服装分类,

.cifar10(100):10个类别分类

python 复制代码
from tensorflow import keras

fashion_mnist=keras.datasets.fashion_mnist

(train_images,train_labels),(test_images,test_labels)=fashion_mnist.load_data()

print(train_images,train_labels)

2.构建模型

在keras中,可以通过组合层来构建模型。模型通常是以层构建成图,最常见模型是层的推叠,keras.layers中就又许多模型:

python 复制代码
from tensorflow.python.keras.layers import Dense
from tensorflow.python.keras.layers import DepthwiseConv2D
from tensorflow.python.keras.layers import Dot
from tensorflow.python.keras.layers import Dropout
from tensorflow.python.keras.layers import ELU
from tensorflow.python.keras.layers import Embedding
from tensorflow.python.keras.layers import Flatten
from tensorflow.python.keras.layers import GRU
from tensorflow.python.keras.layers import GRUCell
from tensorflow.python.keras.layers import LSTMCell
...
...
...

Flatten:将输入数据进行改变展开

Dense:添加一层神经元,Dense(units,activation=None,**kwargs)

units:神经元个数

activation:激活函数

**kwargs:输入上层输入的形状minput_shape=()

(其他方法看上一章)

python 复制代码
model=keras.Sequential([
    # 将输入的数据改变容易训练的形状
    keras.layers.Flatten(input_shape=(28,28)),
    # 设置第一个隐藏
    keras.layers.Dense(128,activation=tf.nn.relu),
    # 设置输出层
    keras.layers.Dense(10,activation=tf.nn.softmax)
])

3.训练与评估

1.通过通过调用model的compile方法去配置该模型所需要的训练参数以及评估方法

2.model.compile(optimizer,loss=None,metrics=None)配置训练相关参数

optimizer:优化器,常用的优化器有:

python 复制代码
from tensorflow.python.keras.optimizers import Adadelta
from tensorflow.python.keras.optimizers import Adagrad
from tensorflow.python.keras.optimizers import Adam
from tensorflow.python.keras.optimizers import Adamax
from tensorflow.python.keras.optimizers import Nadam
from tensorflow.python.keras.optimizers import Optimizer
from tensorflow.python.keras.optimizers import RMSprop
from tensorflow.python.keras.optimizers import SGD
from tensorflow.python.keras.optimizers import deserialize
from tensorflow.python.keras.optimizers import get
from tensorflow.python.keras.optimizers import serialize
from tensorflow.python.keras.optimizers import AdamOptimizer()

loss:损失类型,类型可以是字符串也可以是该损失函数:

python 复制代码
from tensorflow.python.keras.losses import KLD
from tensorflow.python.keras.losses import KLD as kld
from tensorflow.python.keras.losses import KLD as kullback_leibler_divergence
from tensorflow.python.keras.losses import MAE
from tensorflow.python.keras.losses import MAE as mae
from tensorflow.python.keras.losses import MAE as mean_absolute_error
from tensorflow.python.keras.losses import MAPE
from tensorflow.python.keras.losses import MAPE as mape
from tensorflow.python.keras.losses import MAPE as mean_absolute_percentage_error
from tensorflow.python.keras.losses import MSE
from tensorflow.python.keras.losses import MSE as mean_squared_error
from tensorflow.python.keras.losses import MSE as mse
from tensorflow.python.keras.losses import MSLE
from tensorflow.python.keras.losses import MSLE as mean_squared_logarithmic_error
from tensorflow.python.keras.losses import MSLE as msle
from tensorflow.python.keras.losses import binary_crossentropy
from tensorflow.python.keras.losses import categorical_crossentropy
from tensorflow.python.keras.losses import categorical_hinge
from tensorflow.python.keras.losses import cosine
from tensorflow.python.keras.losses import cosine as cosine_proximity
from tensorflow.python.keras.losses import deserialize
from tensorflow.python.keras.losses import get
from tensorflow.python.keras.losses import hinge
from tensorflow.python.keras.losses import logcosh
from tensorflow.python.keras.losses import poisson
from tensorflow.python.keras.losses import serialize
from tensorflow.python.keras.losses import sparse_categorical_crossentropy
from tensorflow.python.keras.losses import squared_hinge

model.fit():进行训练

model.fit(x=None,y=None,batch_size=None,epochs=None,callbacks=None)

x:训练数据

y:目标值

batch_size:每次训练的样本数

epochs:训练多少次

callbacks:回调函数,比如tensorboard

python 复制代码
model=keras.Sequential([
    # 将输入的数据改变容易训练的形状
    keras.layers.Flatten(input_shape=(28,28)),
    # 设置第一个隐藏
    keras.layers.Dense(128,activation=tf.nn.relu),
    # 设置输出层
    keras.layers.Dense(10,activation=tf.nn.softmax)
])

# 设置模型的优化器为Adam,损失为交叉熵损失,metrics为度量的依据,这里是准确率
model.compile(optimizer=tf.keras.optimizers.Adam(),
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])
# 开始模型训练
model.fit(train_images,train_labels,epochs=1)
# 模型评估
model.evaluate(test_images,test_labels)

二.案例:实现多层神经网络进行时装分类

keras中的数据涵盖10个类别(裤子,衬衫,裙子......),70000张灰度图像

1.读取数据

python 复制代码
from tensorflow import keras
import tensorflow as tf


class FashionClassification:

    def __init__(self):
        (self.train,self.train_label),(self.test,self.test_label)=keras.datasets.fashion_mnist.load_data()
        

2.建立模型

python 复制代码
class FashionClassification:
    
    model=keras.Sequential([
        # 将输入的数据转换形状,图片一般使用[28,28]形状
        keras.layers.Flatten(input_shape=(28,28)),
        # 建立第一给隐藏,128个神经元
        keras.layers.Dense(128,activation=tf.nn.relu),
        # 建立输出层,十个输出类别,使用softmax分类
        keras.layers.Dense(10,activation=tf.nn.softmax)
    ])
    
    def __init__(self):
        (self.train,self.train_label),(self.test,self.test_label)=keras.datasets.fashion_mnist.load_data()

3.设置模型参数,开始训练,评估模型

python 复制代码
    def compile(self):
        # 设置模型使用的优化器,设置损失函数
        FashionClassification.model.Compile(optimizer=tf.train.AdamOptimizer(),
                               loss=tf.keras.losses.sparse_categorical_crossentropy,
                               metrics=['accuracy'])
        return None

    def fit(self):
        # 训练函数,训练五次
        FashionClassification.model.fit(self.train,self.train_label,epochs=5)

        return None

    def evaluate(self):
        # 模型评估,
        test_loss,test_acc=FashionClassification.model.evaluate(self.test,self.test_label)
        # 打印损失和准确率
        print(test_loss,test_acc)
        
        return None

4.保存和加载模型

model.save_weights('保存的路径')

model.load_weights('模型所在的路径')

python 复制代码
if __name__ == '__main__':
    # 创建对象
    snn=FashionClassification()
    # 开始训练
    snn.compile()
    snn.fit()
    snn.evaluate()

    # 保存模型
    FashionClassification.model.save_weights("./ckpt/SingleNN.h5")

5.使用保存的模型进行预测

python 复制代码
    def predict(self):
        # 查看是否已经有训练好的数据,有就直接用用
        if os.path.exists('./data/checkpoint'):
            FashionClassification.model.load_weights('./data/FashionClassification')

        predictions=FashionClassification.predict(self.test)

        # print(np.argmax(predictions,1))

        return predictions
python 复制代码
if __name__ == '__main__':
    # # 创建对象
    snn=FashionClassification()
    # # 开始训练
    # snn.compile()
    # snn.fit()
    # snn.evaluate()

    # 保存模型
    # FashionClassification.model.save_weights("./ckpt/SingleNN.h5")
    # 进行模型预测
    predictions = snn.predict()
    # [10000, 10]
    print(predictions)
    print(np.argmax(predictions, axis=1))

6.添加tensorboard

python 复制代码
 def fit(self):

        # 调用tensorboard回调函数
        board = keras.callbacks.TensorBoard(log_dir='"./graph/', write_graph=True)

        # 训练函数,训练五次
        FashionClassification.model.fit(self.train, self.train, epochs=5,
                           batch_size=128,callbacks=[board])


        return None

四.完整代码

python 复制代码
from tensorflow import keras
import tensorflow as tf
import os


class FashionClassification:

    model=keras.Sequential([
        # 将输入的数据转换形状,图片一般使用[28,28]形状
        keras.layers.Flatten(input_shape=(28,28)),
        # 建立第一给隐藏,128个神经元
        keras.layers.Dense(128,activation=tf.nn.relu),
        # 建立输出层,十个输出类别,使用softmax分类
        keras.layers.Dense(10,activation=tf.nn.softmax)
    ])

    def __init__(self):
        # 返回两个元组,测试集和训练集
        # x_train(60000,784),y_train(60000,1)
        (self.train,self.train_label),(self.test,self.test_label)=keras.datasets.fashion_mnist.load_data()

        # 数据归一化,一般图片处理都要归一化,不然效果会很差
        self.train = self.train / 255.0
        self.test = self.test / 255.0
    def compile(self):
        # 设置模型使用的优化器,设置损失函数
        FashionClassification.model.compile(optimizer=keras.optimizers.Adam(),
                               loss=tf.keras.losses.sparse_categorical_crossentropy,
                               metrics=['accuracy'])
        return None

    def fit(self):

        # 调用tensorboard回调函数
        board = keras.callbacks.TensorBoard(log_dir='"./graph/', write_graph=True)

        # 训练函数,训练五次
        FashionClassification.model.fit(self.train, self.train, epochs=5,
                           batch_size=128,callbacks=[board])


        return None

    def evaluate(self):
        # 模型评估,
        test_loss,test_acc=FashionClassification.model.evaluate(self.test,self.test_label)
        # 打印损失和准确率
        print(test_loss,test_acc)

        return None

    def predict(self):
        # 查看是否已经有训练好的数据,有就直接用用
        if os.path.exists('./data/checkpoint'):
            FashionClassification.model.load_weights('./data/FashionClassification')

        predictions=FashionClassification.predict(self.test)

        # print(np.argmax(predictions,1))

        return predictions


if __name__ == '__main__':
    # 创建对象
    snn=FashionClassification()
    # 开始训练
    snn.compile()
    snn.fit()
    snn.evaluate()

    # 保存模型
    # FashionClassification.model.save_weights("./ckpt/SingleNN.h5")
    # 进行模型预测
    # predictions = snn.predict()
    # # [10000, 10]
    # print(predictions)
    # print(np.argmax(predictions, axis=1))

都看到这里了,点个赞把!!!

相关推荐
yolo大师兄1 分钟前
【YOLO系列(V5-V12)通用数据集-火灾烟雾检测数据集】
人工智能·深度学习·yolo·目标检测·机器学习
jndingxin6 分钟前
OpenCV 图形API(15)计算两个矩阵(通常代表二维向量的X和Y分量)每个对应元素之间的相位角(即角度)函数phase()
人工智能·opencv
liruiqiang0522 分钟前
循环神经网络 - 机器学习任务之同步的序列到序列模式
网络·人工智能·rnn·深度学习·神经网络·机器学习
JOYCE_Leo1623 分钟前
图像退化对目标检测的影响 !!
人工智能·目标检测·目标跟踪
IT观察36 分钟前
Spark 2.0携手Solcore:AI重构去中心化质押算力生态 !
人工智能·重构·spark
摆烂仙君39 分钟前
3D意识(3D Awareness)浅析
人工智能·深度学习·计算机视觉·3d
cnbestec1 小时前
Hello Robot创新突破!Stretch3机器人搭载RUMs模型实现未知环境中“即插即用”
人工智能·深度学习·机器人
说私域1 小时前
定制开发开源AI智能名片S2B2C商城小程序:技术赋能商业价值实现路径研究
大数据·人工智能·小程序·开源
肥猪猪爸1 小时前
浅析RAG技术:大语言模型的知识增强之道
人工智能·语言模型·自然语言处理·langchain·大模型·ollama·deepseek
说私域1 小时前
基于开源AI大模型AI智能名片S2B2C商城小程序,探究私域电商中人格域积累与直播电商发展
人工智能·小程序·开源·零售