classifier.py 文件源码

python
阅读 26 收藏 0 点赞 0 评论 0

项目:django_text_classifier 作者: django-text-classifier 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号