INNER CODE UNIT · Python

train_and_predict

dragondrop-cloud/cloud-concierge · nlpengine/main.py:25

def train_and_predict(request):
    """
    Train and predict function. Input request is expected to have a json body with the following
    structure:
    {
        "workspace_docs": {...},
        "new_resource_docs": {...},
    }
    """
    try:
        # Loading data from request
        json_body = request.get_json()
        category_docs = literal_eval(json_body["workspace_docs"])
        new_resource_docs = literal_eval(json_body["new_resource_docs"])

        spacy.util.fix_random_seed(42)
        # Loading spacy model
        nlp = spacy.load("en_core_web_md")

View source record →