时间戳

首页 > 解决方案 > 如何从函数创建服务？

python - 如何从函数创建服务？

问题描述

我想使用 Django Rest API 创建一个服务。我有一个功能。这个函数的结果应该返回 2 个值，我应该以 JSON API 格式返回这些值。该功能将像这样工作。我将接收features_list作为参数，我将使用它来创建结果并在def prediction函数中以 json 格式将其显示为服务。

我创建了一个示例 API（我猜）它class PredictionSet在我的视图中，但我实际上想def prediction在我的视图中提供服务功能。

我无法理解如何应用它。我感到很困惑。任何帮助，将不胜感激。

模型.py

class Liquidity(models.Model):
    pred_y = models.CharField(max_length=600)
    score = models.FloatField()

视图.py

class PredictionSet(viewsets.ModelViewSet):
    queryset = Liquidity.objects.all()
    serializer_class = LiquiditySerializer


def prediction(request, features_list):
    filename = config.FINAL_MODEL_PATH
    classifier = pickle.load(open(filename, 'rb'))
    scale_file = config.SCALER_PATH
    scaler = pickle.load(open(scale_file, 'rb'))

    sample = np.array(features_list).reshape(1, -1)
    sample_scaled = scaler.transform(sample)

    pred_y = classifier.predict(sample_scaled)
    prob_y = classifier.predict_proba(sample_scaled)
    if prob_y[0][1] < 0.5:
        score = 0
    elif prob_y[0][1] <= 0.69:
        score = 1
    else:
        score = 2
    pred_y = pred_y[0]

    prediction_obj = Liquidity.objects.get_or_create(pred_y=pred_y, score=score)
    prediction_result = prediction_obj.pred_y
    prediction_score = prediction_obj.score

    context = {
        'prediction_result ': prediction_result,
        'prediction_score ': prediction_score,
    }

    return context

序列化程序.py

class LiquiditySerializer(serializers.HyperlinkedModelSerializer):
    class Meta:
        model = Liquidity
        fields = '__all__'

标签： pythondjangodjango-rest-framework

解决方案

如果您想从ModelViewsetDRF 中返回自定义 JSON，您可以覆盖.list()和/或.retrieve()像这样：

from rest_framework import status
from rest_framework.response import Response

class PredictionSet(viewsets.ModelViewSet):
    queryset = Liquidity.objects.all()
    serializer_class = LiquiditySerializer
    
    # Your custom function definition
    def prediction(self, request, features_list):
        # The content
    
    def retrieve(self, request, *args, **kwargs):
        result = prediction(...)  # Call your custom service and got result
        # Return the result as JSON (url = /api/v1/predictions/1) an object
        return Response({'data': result}, status=status.HTTP_200_OK)
    
    def list(self, request, *args, **kwargs):
        result = prediction(...)  # Call your custom service and got result
        # Return the result as JSON (url = /api/v1/predictions) a list of objects
        return Response({'data': result}, status=status.HTTP_200_OK)

有关更多详细信息，请点击此链接

推荐阅读