如何解决返回所选测试集的索引Python
我正在尝试获取测试数据选择的数据的索引。首先,我将train-test-split用于数据
A = [[1,2],[3,4],[6,4]]
y = [1,1]
from sklearn.model_selection import train_test_split
A_train,A_test,y_train,y_test = train_test_split(A,y,test_size=0.3,random_state=1)
A_test
>> [[3,2]]
但是,如何获取A中A_test的索引?例如,在这种情况下,它应该返回3、4。非常感谢!
解决方法
只需在Content-Type: text/csv;charset=utf-8;header=present
和range(len(A))
上方使用A
:
y
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。