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)