TensorFlow高阶API使用与PyTorch的安装

最新推荐文章于 2024-07-04 10:02:46 发布

Papicatch

最新推荐文章于 2024-07-04 10:02:46 发布

阅读量1.3k

点赞数 55

分类专栏：专业知识文章标签： tensorflow pytorch 人工智能高阶API python

本文链接：https://blog.csdn.net/2302_76516899/article/details/139909239

版权

专业知识专栏收录该内容

36 篇文章 23 订阅

订阅专栏

欢迎来到 Papicatch的博客

文章目录

🍉高阶API标准化搭建实例：鸢尾花特征分类实验

🍇安装Anaconda或Miniconda

上篇文章为TensorFlow的安装及中低API操作哦，感兴趣的同学可以看一下哦！！！

TensorFlow的安装与使用

🍉TensorFlow高阶API使用

🍈示例1：使用tf.keras构建模型

🍍通过“序贯式”方法构建模型

import tensorflow as tf
from tensorflow.keras import layers

# 构建序贯式模型
model = tf.keras.Sequential([
    layers.Dense(64, activation='relu', input_shape=(4,)),  # 输入层：4个输入节点
    layers.Dense(64, activation='relu'),  # 隐藏层：64个节点
    layers.Dense(3, activation='softmax')  # 输出层：3个输出节点（类别）
])

model.summary()

🍍通过“函数式”方法构建模型

import tensorflow as tf
from tensorflow.keras import layers, Model

# 输入层
inputs = layers.Input(shape=(4,))
# 隐藏层
x = layers.Dense(64, activation='relu')(inputs)
x = layers.Dense(64, activation='relu')(x)
# 输出层
outputs = layers.Dense(3, activation='softmax')(x)

# 构建模型
model = Model(inputs=inputs, outputs=outputs)

model.summary()

🍈示例2：编译模型关键代码

在编译模型时，我们需要指定优化器、损失函数和评估指标。

model.compile(
    optimizer='adam',  # 优化器
    loss='sparse_categorical_crossentropy',  # 损失函数
    metrics=['accuracy']  # 评估指标
)

🍈示例3：训练模型关键代码

在训练模型时，我们使用fit方法。

# 加载鸢尾花数据集
from sklearn.datasets import load_iris
iris = load_iris()
data = iris.data
labels = iris.target

# 分割训练集和测试集
from sklearn.model_selection import train_test_split
train_data, test_data, train_labels, test_labels = train_test_split(data, labels, test_size=0.2, random_state=42)

# 训练模型
model.fit(train_data, train_labels, epochs=50, batch_size=16, validation_split=0.2)

🍉高阶API标准化搭建实例：鸢尾花特征分类实验

🍈实验目标

使用三层的人工神经网络对鸢尾花数据集进行分类。

🍈数据准备

from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
import tensorflow as tf

# 加载数据集
iris = load_iris()
data = iris.data
labels = iris.target

# 分割训练集和测试集
train_data, test_data, train_labels, test_labels = train_test_split(data, labels, test_size=0.2, random_state=42)

🍈构建模型

使用“序贯式”方法构建三层人工神经网络。

from tensorflow.keras import layers

# 构建模型
model = tf.keras.Sequential([
    layers.Dense(64, activation='relu', input_shape=(4,)),
    layers.Dense(64, activation='relu'),
    layers.Dense(3, activation='softmax')
])

# 编译模型
model.compile(
    optimizer='adam',
    loss='sparse_categorical_crossentropy',
    metrics=['accuracy']
)

model.summary()

🍈训练模型

# 训练模型
model.fit(train_data, train_labels, epochs=50, batch_size=16, validation_split=0.2)

🍈评估模型

# 评估模型
test_loss, test_acc = model.evaluate(test_data, test_labels)
print(f"Test Accuracy: {test_acc}")

以上是一个完整的使用TensorFlow高阶API构建、编译和训练神经网络模型的实例，通过对鸢尾花数据集的特征进行分类展示了这些步骤的具体实现。

🍉PyTorch的安装

🍈引言

TensorFlow之后用于深度学习的主要框架是PyTorch。PyTorch框架是Facebook开发的，Twitter和Salesforce等公司都使用PyTorch框架。与TensorFlow不同，PyTorch使用动态更新的图形进行操作，意味着它可以在流程中更改体系结构。在PyTorch中，可以使用标准调试器，如pdb或PyCharm。

PyTorch训练神经网络的过程简单明了，同时，PyTorch支持数据并行和分布式学习模型，还包含很多预先训练的模型。

🍈安装PyTorch

PyTorch可以作为PyTorch包使用，用户可以使用pip或者conda来构建，或者从源码构建等。

🍍使用`conda`安装

🍇安装Anaconda或Miniconda

如果没有安装，可以从 Anaconda官网或 Miniconda官网下载并安装。

🍇创建虚拟环境

为了避免依赖冲突，建议在虚拟环境中安装PyTorch

conda create -n pytorch_env python=3.9
conda activate pytorch_env

🍇选择安装命令：

根据 PyTorch官网提供的配置选择适合的命令。例如：

安装CPU版本

conda install pytorch torchvision torchaudio cpuonly -c pytorch

安装带CUDA支持的版本（例如CUDA 11.7）

conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c nvidia

🍇执行安装命令

在终端或命令提示符中输入上述命令进行安装。

PyTorch官网地址为：Start Locally | PyTorch 。

🍍使用`pip`安装

pip是Python的包管理工具。以下步骤适用于Windows、macOS和Linux。

🍇确保已安装Python和pip

确认已安装Python和pip。可以通过以下命令检查

python --version
pip --version

上图为我电脑安装的Python版本及pip版本。

🍇选择安装命令

根据官方PyTorch网站的推荐，选择适合自己系统和需求的命令。可以访问 PyTorch官网选择具体配置。

以下是一些常见的命令：

安装CPU版本

pip install torch torchvision torchaudio

安装带CUDA支持的版本（例如CUDA 11.7）

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117

🍇执行安装命令

打开终端或命令提示符，输入上述命令进行安装。

🍍通过源码安装

有时需要从源码编译安装PyTorch，适用于自定义需求或开发者。

🍇安装依赖

在安装PyTorch源码之前，需要安装一些必要的依赖项。以Ubuntu为例：

sudo apt-get update
sudo apt-get install cmake git libopenblas-dev liblapack-dev libjpeg-dev libpng-dev

🍇克隆PyTorch源码

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch

🍇配置编译环境

设置Python环境，并确保安装了所需的Python包（如numpy和pillow）。

conda create -n pytorch_from_source python=3.9
conda activate pytorch_from_source
pip install numpy pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses

🍇编译和安装PyTorch

python setup.py install

🍈验证

以上是几种安装PyTorch的方法，根据您的具体需求和系统环境选择合适的方法进行安装。安装完成后，可以通过以下代码测试安装是否成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果安装了CUDA支持的版本，检查CUDA是否可用

Papicatch

关注

55
点赞
踩
28

收藏

觉得还不错? 一键收藏
32
评论
TensorFlow高阶API使用与PyTorch的安装

以上是一个完整的使用TensorFlow高阶API构建、编译和训练神经网络模型的实例，通过对鸢尾花数据集的特征进行分类展示了这些步骤的具体实现。以上是几种安装PyTorch的方法，根据您的具体需求和系统环境选择合适的方法进行安装。PyTorch训练神经网络的过程简单明了，同时，PyTorch支持数据并行和分布式学习模型，还包含很多预先训练的模型。PyTorch可以作为PyTorch包使用，用户可以使用pip或者conda来构建，或者从源码构建等。在安装PyTorch源码之前，需要安装一些必要的依赖项。
复制链接

扫一扫