首页 > 解决方案 > 训练测试拆分 Fastai

问题描述

每次我要使用 keras 进行训练时,我都使用 numpy.random.seed() 以相同的顺序拆分数据集。

当我尝试 fastai 时,我有一个疑问,numpy.random.seed() 函数是否会与 keras 中的一样工作,我使用 ImageDataBunch.from_folder() 函数来加载数据集......

如果没有,每次我在 fastai 中加载数据集时,如何以相同的顺序拆分数据集

谢谢

标签: pythonmachine-learningdeep-learningpytorchfast-ai

解决方案


根据文档,您可以使用from_folder()一个seed论点:

ImageDataBunch.from_folder(seed=1234)

推荐阅读