首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >train_test_split函数的分层python包装3

train_test_split函数的分层python包装3
EN

Stack Overflow用户
提问于 2022-05-20 16:17:16
回答 1查看 34关注 0票数 0

尝试做一个train_test_split,但数据库是如此的不平衡,所以我想分层它,我怎么做呢?

在文档中,我看到train_test_split函数只接收两个参数:

Train_test_split(百分比,rnd=None)

所以不知道分层是否有可能做到这一点。

我的代码是:

代码语言:javascript
运行
复制
train_model_1_ORG, test_model_1_ORG = data_modelos_1_2.train_test_split(70.0, Random(1))
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2022-05-21 23:39:59

不,Weka不提供这种功能。

但是,我只是向weka.core.Instances类提交了一个更改,允许您生成交叉验证拆分(方法:cv_splits)。

由于您希望在您的培训集70%,您可以生成一个10倍的简历分裂,然后将7个测试分裂成一个训练集和其他的一个测试集(使用Instances.append_instances(Inst1, Inst2)类方法)。

NB: --您需要安装pww3 直接从吉乌布或使用比0.2.9更新的版本。

票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/72321968

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档