我正在尝试从给定文档中提取一些特征,给定一组预定义的特征.
from sklearn.feature_extraction.text import CountVectorizer
features = ['a', 'b', 'c']
doc = ['a', 'c']
vectoriser = CountVectorizer()
vectoriser.vocabulary = features
vectoriser.fit_transform(doc)
但是输出是一个2x3数组,用零填充而不是:
desired_output = [[1, 0, 0]
[0, 0, 1]]
任何帮助将非常感激