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 参数指定新类别的类别标签。在模型推理时,我们需要将新的类别标签与之前的类别标签合并,并将输出层的大小增加以适应新的类别数量。

相关内容

热门资讯

现有说明如下!德普之星透视免费... 现有说明如下!德普之星透视免费(透视)一贯真的是有挂(真是有挂开挂辅助脚本)-哔哩哔哩【无需打开直接...
近年来!pokernow辅助工... 近年来!pokernow辅助工具(透视)一直是有挂(有挂头条开挂辅助下载)-哔哩哔哩;无需打开直接搜...
现就发布提示!wepoker数... 现就发布提示!wepoker数据分析工具(透视)果然真的是有挂(存在有挂开挂辅助脚本)-哔哩哔哩1、...
今天下午!epoker免费透视... epoker免费透视脚本是一款专注玩家量身打造的游戏记牌类型软件,在epoker免费透视脚本这款游戏...
事发当天!hhpoker德州真... 事发当天!hhpoker德州真的假的(透视)其实真的是有挂(果真有挂开挂辅助安装)-哔哩哔哩>>您好...
一直以来!wepoker免费辅... 一直以来!wepoker免费辅助器(透视)果然是有挂的(今日头条开挂辅助神器)-哔哩哔哩【无需打开直...
突发!德普之星怎么设置埋牌(透... 突发!德普之星怎么设置埋牌(透视)原来是有挂(有挂技术开挂辅助辅助器)-哔哩哔哩德普之星怎么设置埋牌...
随着!aapoker怎么选牌(... 随着!aapoker怎么选牌(透视)一直确实有挂(有挂教程开挂辅助平台)-哔哩哔哩您好:aapoke...
此事引发网友热议!wpk有辅助... 此事引发网友热议!wpk有辅助吗(透视)一直确实有挂(有挂方法开挂辅助安装)-哔哩哔哩;无需打开直接...
今天上午!约局吧德州真的存在透... 约局吧德州真的存在透视吗是一款可以让一直输的玩家,快速成为一个“必胜”的ai辅助神器,有需要的用户可...