如何使用scikitlearn交叉验证约束

2024-03-28 16:48:04 发布

您现在位置:Python中文网/ 问答频道 /正文

输入:

  • 我有一个图像数组X,其中每一行都是代表一个人的示例。你知道吗
  • 另一个数组y用于它们的标签,其中标签是1到7之间的整数。你知道吗
  • 以及ids的最后一个数组,其中ids[i]表示X[i]处第i个人的id。(同一个人有同一个id,同一个人可能有不同的图像。)

有没有可能对Xy进行分区,这样同一个人就不会同时进入测试和训练集中?你知道吗

我想我需要使用sklearn.cross_validation.train_test_split。有人能解释一下“分层”是做什么的吗?这是我想做的正确方法吗?你知道吗


Tags: test图像idids示例train代表整数