Python Функция для включения преобразования данных - PullRequest
0 голосов
/ 03 марта 2020

Я новичок в python, и я разработал модель и хотел бы создать функцию для использования API. Сложная часть функции заключается в том, чтобы превратить ввод данных (когда кто-то запрашивает) в pd.get_dummies (70 столбцов)

Ожидаемым параметром для запроса пользователя является A, B, C, D , E, F, G. с помощью макетов pd.get он преобразует данные в двоичные данные в столбцы A1, A2, A3 ... A28, B1, B2 ... B4, C1, C2..C4, D1, D2 и D3, E1, E2 .... E13, F1, F2 ... F5, G1, G2 ... G13.

Например, если A = 2, манекены pd.get поставят 1 в столбец A2, а оставив A1, а остальные - 0. Может кто-нибудь помочь с моим кодом ниже? Заранее спасибо!

import pandas as pd
import pickle

# load model
model = pickle.load(open('model.pkl','rb'))

def predict(Project_Sector, IncidentType, Parent_Child_Type, Project_Lifecycle_Phase, LL_BU, Project_Size, HeadContractType):

    features = ['Project_Sector', 'IncidentType', 'Parent_Child_Type',
       'Project_Lifecycle_Phase', 'LL_BU', 'Project_Size', 'HeadContractType']

    s = predict("Project_Sector", "IncidentType", "Parent_Child_Type", "Project_Lifecycle_Phase", "LL_BU", 'Project_Size", "HeadContractType")

    data_transf = pd.get_dummies(s,columns=features)

    # predictions
    result = model.predict(data_transf)

    # send back to browser
    output = {'results': int(result[0])}

    # return data
    return jsonify(results=output)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...