过度拟合ML模型

时间:2019-12-08 15:15:58

标签: machine-learning computer-vision classification

我的图像分类器有问题,它显示了超过500%的验证错误,并且给我2%的训练损失以及90%+的训练精度。

我还尝试使用 fastai resnet50 进行迁移学习,它也给我相同的结果。

这是我的结果: -验证错误:500% -验证错误:大约2% -验证准确度:90%+ -没有纪元:50

提示:我的数据是从Google抓取的图像,我有10个班级,每个班级约有600-700张图像。

感谢您的时间

这是我的代码专家。

from fastai.vision import *
from fastai.metrics import error_rate

path = Path('drive/My Drive/cheese images/')

np.random.seed(42)
data = ImageDataBunch.from_folder(path, train='.', valid_pct = 0.2,
                              ds_tfms = get_transforms(), size = 224, 
num_workers=4).normalize(imagenet_stats)

learn = cnn_learner(data, models.resnet50, metrics = accuracy)

learn.fit_one_cycle(50)
learn.export()
learn.unfreeze()

0 个答案:

没有答案