Path: blob/master/model_deployment/fastapi_kubernetes/app/__pycache__/main.cpython-36.pyc
1480 views
3 ��^� � @ s� d dl mZ d dlmZ d dlmZ d dlZd dlZ e� Z e j dd�Z G dd� de�Z G d d � d e�ZG d d � d e�Ze jd �dd� �Ze jded�e d�dd��Ze jded�ee d�dd��ZdS )� )�List)�FastAPI)� BaseModelNz./app/model.txt)Z model_filec @ s. e Zd ZU eeeeee e e dS )�FeaturesN) �__name__� __module__� __qualname__�floatZMedIncZHouseAgeZAveRoomsZ AveBedrmsZ PopulationZAveOccupZLatitudeZ Longitude� r r � ./app/main.pyr s r c @ s e Zd ZU edS )�PredictResponseN)r r r r �scorer r r r r s r c @ s e Zd ZU ee dS )�BatchPredictResponseN)r r r r r �scoresr r r r r s r �/c C s ddiS )N�messagez Hello Worldr r r r r � hello_world s r z/predict)Zresponse_model)�featuresc s> t j� fdd�tj� D ��}|jdd�}tj|�d }d|iS )z3Endpoint for getting the score for a single record.c s g | ]}t � |��qS r )�getattr)�.0�name)r r r � <listcomp>& s zpredict.<locals>.<listcomp>� r r �����)�np�array�model� feature_nameZreshape�predict)r �rowr r )r r r # s r z/batch/predict)�batch_featuresc sn t tj� �}tjt | �|f�}x8t| �D ],\}� tj� fdd�tj� D ��}|||< q(W tj|�j� }d|iS )z2Endpoint for getting scores for a batch of record.c s g | ]}t � |��qS r )r )r r )r r r r 6 s z!batch_predict.<locals>.<listcomp>r ) �lenr r r Zzeros� enumerater r �tolist)r Z num_features�rows�ir r r )r r � batch_predict/ s r& )�typingr Zfastapir Zpydanticr Znumpyr ZlightgbmZlgb�appZBoosterr r r r �getr Zpostr r& r r r r �<module> s