Я новичок в 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)