def predict():
"""
予測テスト
"""
result = None
if request.method == "POST":
article = request.form['article']
nlp = Nlp()
classes = nlp.predict([article])
c = classes[0][0]
result = {'article': article, 'class': c}
return render_template("predict.html", result=result)
def index(label=None):
"""
記事一覧表示
"""
db.build_db()
limit = 0
if request.args.get('label'):
label = int(request.args.get('label'))
labels = (label,)
else:
label = 'all'
labels = (0, 1, 2)
articles = db.get_articles(labels, limit)
# 予測
nlp = Nlp()
predict_labels = [pl[0] for pl in nlp.predict([a[3] for a in articles])]
return render_template("index.html", articles=articles, predict_labels=predict_labels, label=label)