Machine Learning Course Cases Based on MindSpore

概述

本项目是基于MindSpore的机器学习案例,共包含34个实验,覆盖机器学习从简单到复杂的各种实验,所有实验使用MindSpore框架实现。

安装

为了运行我们的代码,需要安装一些运行包

pip install -r requirements.txt

硬件环境

本项目是基于MindSpore2.0实现的,在昇腾1.8上实现的,部分实验要求内存大于32GB,显存大于16GB。 本项目所有实验均可在GPU、CPU、Ascend上运行。

运行

下面是本项目运行的一些说明

1. 数据集准备

MNIST, CIFAR-10, CIFAR-100,鸢尾花数据集 可在代码运行的过程中直接下载,ImageNet数据集需在官方网站进行下载。

2. 实验运行

本项目中的每个实验均包含主运行文件,在每个实验中的notebook文件中,对该实验中的每一个步骤均有详细的说明,只需按照说明,运行notebook文件即可。

3. 结果展示

本项目中的实验均包含可视化结果展示

致谢

本项目为智能计算技术研究中心团队及华为公司共同构建,在项目完成的过程中,参考了符合MIT许可的部分代码,由于涉及人数较多,不再一一详述,在此一并感谢。