示例#1
0
def classify_news():
    s = session()
    rows = s.query(News).filter(News.label == None).all()

    model = NaiveBayesClassifier()
    model.import_model('news_model.json')
    predictions = model.predict([row.title for row in rows])

    d = deque()
    for new, pred in zip(rows, predictions):
        if pred == 'good':
            d.appendleft(new)
        elif pred == 'never':
            d.append(new)

    return template('classify', rows=d)