如何按组拆分数据?

尤金

我有以下数据

user_id, item_id
1, 2
1, 3
2, 5,
2, 4
1, 4

我想用约束分割数据:

每个用户都必须使用训练集或测试集,但不能同时使用

因此[[1,2], [1,3], [1,4]] / [[2,5],[2,4]]是有效拆分,
[[1,2],[2,5][1,4]] / [[1,3]... ]不是有效拆分,因为用户已拆分

我不确定是否from sklearn.model_selection import train_test_split支持它。

叶夫根尼·波波维奇(Evgeny Popovich)

如果您希望培训和测试中有固定比例的用户,则需要获得唯一用户和拆分用户。下面的伪代码

train_ids, test_ids = train_test_split(unique_users)
train_X = data[data.user_id==train_ids]
test_X = data[data.user_id==test_ids]

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章