samples = [['asdf', '1'], ['asdf', '0']]
# turn the samples into dicts
samples = [dict(enumerate(sample)) for sample in samples]
# turn list of dicts into a numpy array
vect = DictVectorizer(sparse=False)
X = vect.fit_transform(samples)
clf = DecisionTreeClassifier()
clf.fit(X, ['2', '3'])
您需要在NumPy数组中将字符串值特征转换为数字特征;^{} 可以为您完成这项工作。在
在将测试样本转换为dict之后,请记住在测试样本上使用
vect.transform
。在相关问题 更多 >
编程相关推荐