gpt4 book ai didi

python - 非整数类标签 Scikit-Learn

转载 作者:IT老高 更新时间:2023-10-28 20:45:49 26 4
gpt4 key购买 nike

scikit-learn 的快速 SVM 问题。当你训练一个 SVM 时,它类似于

from sklearn import svm
s = svm.SVC()
s.fit(training_data, labels)

有没有办法让 labels 成为非数字类型的列表?例如,如果我想将向量分类为“猫”或“狗”,而不必使用某种外部查找表将“猫”和“狗”编码为 1 和 2。当我尝试只传递一个字符串列表时,我得到...

ValueError:float() 的无效文字:cat

所以,看起来只是在 labels 中插入字符串会起作用。有什么想法吗?

最佳答案

直接将字符串作为类传递是我的待办事项,但 SVM 尚不支持它。目前,我们有 LabelEncoder可以帮你记账。

[edit]这应该可以立即使用[/edit]

关于python - 非整数类标签 Scikit-Learn,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/13300160/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com