本项目是基于MindSpore的机器学习案例,共包含34个实验,覆盖机器学习从简单到复杂的各种实验,所有实验使用MindSpore框架实现。
为了运行我们的代码,需要安装一些运行包
pip install -r requirements.txt
本项目是基于MindSpore2.0实现的,在昇腾1.8上实现的,部分实验要求内存大于32GB,显存大于16GB。 本项目所有实验均可在GPU、CPU、Ascend上运行。
下面是本项目运行的一些说明
MNIST, CIFAR-10, CIFAR-100,鸢尾花数据集 可在代码运行的过程中直接下载,ImageNet数据集需在官方网站进行下载。
本项目中的每个实验均包含主运行文件,在每个实验中的notebook文件中,对该实验中的每一个步骤均有详细的说明,只需按照说明,运行notebook文件即可。
本项目中的实验均包含可视化结果展示
本项目为智能计算技术研究中心团队及华为公司共同构建,在项目完成的过程中,参考了符合MIT许可的部分代码,由于涉及人数较多,不再一一详述,在此一并感谢。