com.graphaware.nlp.dsl.request.CustomModelsRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of nlp Show documentation
GraphAware Framework Module for adding NLP capabilities to Neo4j.
There is a newer version: 3.5.4.53.18
/*
 * Copyright (c) 2013-2018 GraphAware
 *
 * This file is part of the GraphAware Framework.
 *
 * GraphAware Framework is free software: you can redistribute it and/or modify it under the terms of
 * the GNU General Public License as published by the Free Software Foundation, either
 * version 3 of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of
 * the GNU General Public License along with this program.  If not, see
 * .
 */
package com.graphaware.nlp.dsl.request;

import org.neo4j.logging.Log;
import com.graphaware.common.log.LoggerFactory;

import java.util.Map;

public class CustomModelsRequest {

    private static final Log LOG = LoggerFactory.getLogger(TextRankRequest.class);

    private String textProcessor;
    private String alg;
    private String modelId;
    private String inputFile;
    private String lang;

    private Map trainingParams;

    private final static String PARAMETER_TEXT_PROCESSOR = "textProcessor";
    private final static String PARAMETER_ALG = "alg";
    private final static String PARAMETER_MODEL_ID = "modelIdentifier";
    private final static String PARAMETER_INPUT_FILE = "inputFile";
    private final static String PARAMETER_LANG = "lang";
    private final static String PARAMETER_TRAINING_PARAMS = "trainingParameters";

    private static final String DEFAULT_LANG = "en";

    public static CustomModelsRequest fromMap(Map request) {
        if (!request.containsKey(PARAMETER_TEXT_PROCESSOR))
            throw new RuntimeException("Missing parameter " + PARAMETER_TEXT_PROCESSOR);
        if (!request.containsKey(PARAMETER_ALG))
            throw new RuntimeException("Missing parameter " + PARAMETER_ALG);
        if (!request.containsKey(PARAMETER_MODEL_ID))
            throw new RuntimeException("Missing parameter " + PARAMETER_MODEL_ID);
        if (!request.containsKey(PARAMETER_INPUT_FILE))
            throw new RuntimeException("Missing parameter " + PARAMETER_INPUT_FILE);

        CustomModelsRequest result = new CustomModelsRequest();
        result.setTextProcessor((String) request.get(PARAMETER_TEXT_PROCESSOR));
        result.setAlg((String) request.get(PARAMETER_ALG));
        result.setModelID((String) request.get(PARAMETER_MODEL_ID));
        result.setInputFile((String) request.get(PARAMETER_INPUT_FILE));
        result.setLanguage((String) request.getOrDefault(PARAMETER_LANG, DEFAULT_LANG));

        if (request.containsKey(PARAMETER_TRAINING_PARAMS)) {
            result.setTrainingParameters((Map) request.get(PARAMETER_TRAINING_PARAMS));
        }

        return result;
    }

    public String getTextProcessor() {
        return this.textProcessor;
    }
    
    private void setTextProcessor(String processor) {
        this.textProcessor = processor;
    }

    public String getAlg() {
        return this.alg;
    }

    private void setAlg(String alg) {
        this.alg = alg;
    }

    public String getModelID() {
        return this.modelId;
    }

    private void setModelID(String id) {
        this.modelId = id;
    }

    public String getInputFile() {
        return this.inputFile;
    }

    private void setInputFile(String file) {
        this.inputFile = file;
    }

    public String getLanguage() {
        return this.lang;
    }

    private void setLanguage(String lang) {
        this.lang = lang;
    }

    public Map getTrainingParameters() {
        return this.trainingParams;
    }

    private void setTrainingParameters(Map params) {
        this.trainingParams = params;
    }

}