在Scikit的2D列表上安装培训标签--学习

Question

我试图将2d中的行映射到带有Scikit-learn的标签列表中的元素。

例如：

​

from sklearn import tree
clf = DecisionTreeClassifier()

#2D list of training data:
training_data = [[1, 2, 3], [1, 2, 4, 5, 6], [5, 7], [1, 2, 3]]

#1D list of training labels:
training_labels = ['a', 'b', 'c', 'a']

clf = clf.fit(training_data, training_labels)

​

当我运行代码时，我得到"ValueError:设置带有序列的数组元素“。

我想知道如何正确地转换数据，以便我可以将测试数据与培训标签相匹配。

Answer 1

testing_data = [[1, 2, 3], [1, 2, 4, 5, 6], [5, 7], [1, 2, 3]]

在这里，如果每个子列表都被认为是一个样本，那么每个样本都没有相同的维度。在这种情况下，不可能对模型进行拟合。

也许你的意思是：

 training_labels = ["a", "b", "c", "a"]

否则，应该定义a,b,c变量。