Sagemaker内置Random Cut Forest算法训练添加验证集报错咨询
SageMaker内置Random Cut Forest(RCF)验证配置解决方案
报错根因
SageMaker官方内置的RCF算法仅预定义了train、test两个合法输入通道,不存在validation通道配置项,因此在fit方法中传入validation字段时会触发输入校验错误。
可行验证实现方案
方案1:使用test通道完成训练期验证(推荐)
官方内置RCF支持传入test通道数据,训练完成后会自动基于该数据集计算异常检测的精确率、召回率、F1值等标准评估指标,完全可以替代validation通道的验证需求,配置示例如下:
注意:test通道的带标签CSV需要满足第一列为标签列(异常标记为1,正常标记为0),后续特征列顺序、维度和训练集完全一致。# train为无标签训练数据S3路径,test传入原本的带标签验证集S3路径 rcf_estimator.fit( inputs={ "train": "s3://your-bucket/path/to/train-data/", "test": "s3://your-bucket/path/to/validation-data/" } )方案2:离线自定义验证
如果需要自定义验证逻辑(比如调整异常分数阈值、计算业务自定义指标),可以先仅用train通道完成训练,之后通过临时端点或者批量转换任务对验证集做推理,自行比对标签计算验证指标,符合预期后再正式部署模型。
内容的提问来源于stack exchange,提问作者Christina Riehn
相关产品推荐
相关产品推荐

