Function predict

app/server.py:16–30

Predicts the class of a given set of features. Args: data (dict): A dictionary containing the features to predict. e.g. {"features": [1, 2, 3, 4]} Returns: dict: A dictionary containing the predicted class.

(data: dict)

@app.post('/predict')
def predict(data: dict):
    """
    Predicts the class of a given set of features.

    Args:
        data (dict): A dictionary containing the features to predict.
        e.g. {"features": [1, 2, 3, 4]}

    Returns:
        dict: A dictionary containing the predicted class.
    """        
    features = np.array(data['features']).reshape(1, -1)
    prediction = model.predict(features)
    class_name = class_names[prediction][0]
    return {'predicted_class': class_name}