ML functions
 
Loading...
Searching...
No Matches
HuggingFaceServerless.h File Reference

Implementation of a Hugging Face serverless API integration for machine learning tasks. More...

#include <cpr/cpr.h>
#include <json/json.h>
#include <nlohmann/json.hpp>
#include <Eigen/Dense>
#include <cmath>
#include <iostream>
#include "BaseFunction.h"
#include "velox/exec/tests/utils/AssertQueryBuilder.h"
#include "velox/exec/tests/utils/PlanBuilder.h"
#include "velox/exec/tests/utils/TempDirectoryPath.h"
#include "velox/ml_functions/UtilFunction.h"
#include "velox/vector/tests/utils/VectorTestBase.h"

Go to the source code of this file.

Classes

class  HuggingFaceServerless
 Implements a machine learning function that interacts with Hugging Face's serverless API. More...
 

Enumerations

enum  HuggingFaceTaskType { TEXT_CLASSIFICATION , IMAGE_CLASSIFICATION , REGRESSION , TEXT_FEATURE_EXTRACTION }
 Enumeration of supported Hugging Face task types. More...
 

Detailed Description

Implementation of a Hugging Face serverless API integration for machine learning tasks.

Enumeration Type Documentation

◆ HuggingFaceTaskType

Enumeration of supported Hugging Face task types.

Enumerator
TEXT_CLASSIFICATION 

Text classification task.

IMAGE_CLASSIFICATION 

Image classification task.

REGRESSION 

Regression task.

TEXT_FEATURE_EXTRACTION 

Text feature extraction task.