Function fit

App/main.py:139–228

()

@app.route('/api-fit', methods = ['POST', 'GET'])
def fit() :

    if request.method == 'POST':

        data = json.loads(str(request.data, 'utf-8'))

        dataSet, discrete, textColumn = pullDataSet(str(data['hashKey']))

        model = -1

        model_id = 0

        ssler = None

        target = ''

        if(dataSet is None) : return '{"status_code" : 403,"msg" : "数据已过期"}'

        if('model' in data) : model_id = model = data['model']

        if(model <= 0 or model > 6) : return '{"status_code" : 403,"msg" : "模型种类无效"}'

        if('target' in data and data['target'] in dataSet.columns): target = data['target']
        else : return '{"status_code" : 403,"msg" : "标签无效"}'

        if(model == 1) :

            y = np.array(dataSet[target])

            model, ssler = NBayesTraining(dataSet, y, textColumn)

        elif(model == 2) :

            x, y = DataFrame2NPArray(dataSet, target)

            model, ssler = KNNTraining(x, y)

        elif(model == 3) :

            x, y = DataFrame2NPArray(dataSet, target)

            model, ssler = SVMTraining(x, y)

        elif(model == 4) :

            x, y = DataFrame2NPArray(dataSet, target)

            model, ssler = LinearRegressionTraining(x, y)

        elif(model == 5) :

            x, y = DataFrame2NPArray(dataSet, target)

            model, ssler = LogisticRegressionTraining(x, y)

        elif(model == 6) :

            x, y = DataFrame2NPArray(dataSet, target)