过度拟合ML模型[重新发布]

时间:2019-12-10 20:46:10

标签: 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 个答案:

没有答案