AWSSagemaker增量训练:是否应该在训练集中包括所有类别?
创始人
2024-09-27 02:00:40
0

在增量训练中,是否应该包括所有类别取决于具体的应用场景。如果一个新的类别加入了训练集,我们可以选择重新训练所有类别,或者只训练新的类别。如果只训练新的类别,那么在模型推理时需要将新的类别与之前的类别合并。下面是一个根据数据集增加新类别的示例代码:

# 加载原始训练集
train_data = sagemaker_session.upload_data(path='train_data', key_prefix='sagemaker/train')

# 加载增量训练集,包括新类别的数据
incremental_data = sagemaker_session.upload_data(path='incremental_data', key_prefix='sagemaker/train', extra_args={'--class-label': 'new_class'})

# 训练模型,加入新类别
estimator = sagemaker.estimator.Estimator(image_name=image_name,
                                          role=role,
                                          train_instance_count=1,
                                          train_instance_type='ml.p3.2xlarge',
                                          output_path=output_path)
estimator.fit({'incremental': incremental_data, 'train': train_data})

在这个示例中,当我们上传增量训练集时,通过 extra_args 参数指定新类别的类别标签。在模型推理时,我们需要将新的类别标签与之前的类别标签合并,并将输出层的大小增加以适应新的类别数量。

相关内容

热门资讯

德州ai辅助!aapoker插... 德州ai辅助!aapoker插件(透视)从来真的是有挂(详细辅助解密教程)是一款可以让一直输的玩家,...
wepoke有没有挂!wepo... 您好,wepoke有没有挂这款游戏可以开挂的,确实是有挂的,需要了解加微【136704302】很多玩...
wepoke模拟器!德扑之星内... wepoke模拟器!德扑之星内部(透视)真是是有挂(详细辅助曝光教程)1、许多玩家不知道wepoke...
微扑克辅助软件!wpk辅助nz... 微扑克辅助软件!wpk辅助nzt(透视)好像是真的有挂(详细辅助详细教程)1)微扑克辅助软件辅助挂:...
aapoker辅助工具!轰趴十... aapoker辅助工具!轰趴十三水有后台吗(透视)切实真的是有挂(详细辅助新版2025教程)是一款可...
微扑克有辅助挂!德扑ai自定义... 微扑克有辅助挂!德扑ai自定义设置数据(透视)其实真的有挂(详细辅助解密教程)是一款可以让一直输的玩...
德州之星有辅助挂!扑克世界牌局... 德州之星有辅助挂!扑克世界牌局设置(透视)一直存在有挂(详细辅助曝光教程)是一款可以让一直输的玩家,...
wpk德州透视辅助!aa po... wpk德州透视辅助!aa poker辅助软件(透视)一贯存在有挂(详细辅助透明教程);小薇(透视辅助...
aapoker透明挂!鱼扑克f... 自定义aapoker透明挂系统规律,只需要输入自己想要的开挂功能,一键便可以生成出微扑克专用辅助器,...
wpk有透视辅助!德州ai辅助... wpk有透视辅助!德州ai辅助工具购买(透视)原先真的有挂(详细辅助存在挂教程)1、wpk有透视辅助...