def get_pipeline(name):
x = TrainingSet.objects.filter(classifier=name).values_list('body',
flat=True)
y = TrainingSet.objects.filter(classifier=name).values_list('target',
flat=True)
pipeline = Pipeline([
('vector', CountVectorizer()),
('transform', TfidfTransformer()),
('bayes', MultinomialNB())
])
pipeline.fit(x, y)
return pipeline
classifier.py 文件源码
python
阅读 26
收藏 0
点赞 0
评论 0
评论列表
文章目录