如何解决为排名模型创建Catboost池时出错
我正在尝试使用Catboost库构建排名模型。在训练集上创建一个池时,出现以下错误。
CatBoostError: catboost/private/libs/data_types/query.cpp:25: Error: queryIds should be grouped
我用来对行进行分组的列是一个字符串数组列。在我看到的示例中,它们大多是整数数组。以前有人遇到过类似的问题吗?
解决方法
我有同样的问题 使用以下代码时
train_data = Pool(
data=input_data[['input_col']],label=input_data[['score']],group_id=input_data[['query_id']],text_features=text_features
)
使用 input_data
对 query_id
排序后得到解决
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。