Ejemplo n.º 1
0
def register_prediction_feedback():
    try:
        sample_text = request.args.get("sample_text", None, type=str)
        sample_prediction = request.args.get("sample_prediction",
                                             None,
                                             type=float)
        sample = Sample(sample_text, prediction_score=sample_prediction)
        correct = request.args.get("correct", None, type=bool)
    except:
        return jsonify(success=False)
    else:
        if correct:
            if sample_prediction > 0.5:
                sample.label = 1
            else:
                sample.label = 0
        else:
            if sample_prediction > 0.5:
                sample.label = 0
            else:
                sample.label = 1
        sample.labeled = True
        push_sample_to_firebase(sample)
        # train_single(sample.text, sample.label)
        return jsonify(success=True)