第八课：ShuffleNet v1、ShuffleNet v2学习

前言

随着人工智能的不断发展，机器学习这门技术也越来越重要，很多人都开启了学习机器学习，本文就介绍了机器学习的基础内容。来源于哔哩哔哩博主“霹雳吧啦Wz”，博主学习作为笔记记录，欢迎大家一起讨论学习交流。

一、搭建ShuffleNet网络

用ShuffleNet v1学习,轮次训练 轮次（代码），个epoch大概 分钟。

二、代码部分

1.module.py----定义ShuffleNet的网络结构

代码如下（示例）：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import warnings
warnings.filterwarnings('ignore')
import  ssl
ssl._create_default_https_context = ssl._create_unverified_context

2.train.py----加载数据集并进行训练，训练集计算loss，测试集计算accuracy，保存训练好的网络参数

代码如下（示例）：

data = pd.read_csv(
    'https://labfile.oss.aliyuncs.com/courses/1283/adult.data.csv')
print(data.head())

3.predict.py——得到训练好的网络参数后，用自己找的图像进行分类测试

在这里插入代码片

三、ShuffleNet v1

四、ShuffleNet v2