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

相关内容

热门资讯

事发当天!上饶辅助设备出租,真... 事发当天!上饶辅助设备出租,真是是真的辅助工具(有挂头条)-哔哩哔哩上饶辅助设备出租脚本下载中分为三...
随着!菠萝神辅助器app,一直... 随着!菠萝神辅助器app,一直是有辅助平台(有挂教程)-哔哩哔哩1、很好的工具软件,可以解锁游戏的菠...
推出新举措!兴动海满辅助,一贯... 您好,兴动海满辅助这款游戏可以开挂的,确实是有挂的,需要了解加去威信【136704302】很多玩家在...
做出回应!微乐贵阳捉鸡麻将挂软... 做出回应!微乐贵阳捉鸡麻将挂软件,果然真的是有辅助神器(有挂秘诀)-哔哩哔哩1、打开软件启动之后找到...
反观!九游破解辅助插件,都是真... 反观!九游破解辅助插件,都是真的有辅助神器(有挂秘籍)-哔哩哔哩九游破解辅助插件能透视中分为三种模型...
据文件显示!科米台州麻将辅助,... 据文件显示!科米台州麻将辅助,真是有挂辅助app(竟然有挂)-哔哩哔哩1、每一步都需要思考,不同水平...
今天上午!赣湘互娱挂,总是是有... 今天上午!赣湘互娱挂,总是是有辅助修改器(有挂详情)-哔哩哔哩1、在赣湘互娱挂插件功能辅助器技巧中,...
来临!广西友乐解码器辅助器,原... 来临!广西友乐解码器辅助器,原来真的是有辅助脚本(真的有挂)-哔哩哔哩1.广西友乐解码器辅助器 选牌...
来临!新天道辅助脚本,确实有挂... 来临!新天道辅助脚本,确实有挂辅助下载(有挂秘诀)-哔哩哔哩1、金币登录送、破产送、升级送、活动送。...
近期!青橙竞技卡五星辅助,好像... 近期!青橙竞技卡五星辅助,好像真的是有辅助脚本(竟然有挂)-哔哩哔哩1、用户打开应用后不用登录就可以...