svm_api.h
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2019-2025, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 #pragma once
6 
7 #include <cuml/cuml_api.h>
8 
9 #ifdef __cplusplus
10 extern "C" {
11 #endif
12 
13 typedef enum cumlSvmKernelType { LINEAR, POLYNOMIAL, RBF, TANH } cumlSvmKernelType;
14 
53 cumlError_t cumlSpSvcFit(cumlHandle_t handle,
54  float* input,
55  int n_rows,
56  int n_cols,
57  float* labels,
58  float C,
59  float cache_size,
60  int max_iter,
61  int nochange_steps,
62  float tol,
63  int verbosity,
64  cumlSvmKernelType kernel,
65  int degree,
66  float gamma,
67  float coef0,
68  int* n_support,
69  float* b,
70  float** dual_coefs,
71  float** x_support,
72  int** support_idx,
73  int* n_classes,
74  float** unique_labels);
75 
76 cumlError_t cumlDpSvcFit(cumlHandle_t handle,
77  double* input,
78  int n_rows,
79  int n_cols,
80  double* labels,
81  double C,
82  double cache_size,
83  int max_iter,
84  int nochange_steps,
85  double tol,
86  int verbosity,
87  cumlSvmKernelType kernel,
88  int degree,
89  double gamma,
90  double coef0,
91  int* n_support,
92  double* b,
93  double** dual_coefs,
94  double** x_support,
95  int** support_idx,
96  int* n_classes,
97  double** unique_labels);
129 cumlError_t cumlSpSvcPredict(cumlHandle_t handle,
130  float* input,
131  int n_rows,
132  int n_cols,
133  cumlSvmKernelType kernel,
134  int degree,
135  float gamma,
136  float coef0,
137  int n_support,
138  float b,
139  float* dual_coefs,
140  float* x_support,
141  int n_classes,
142  float* unique_labels,
143  float* preds,
144  float buffer_size,
145  int predict_class);
146 
147 cumlError_t cumlDpSvcPredict(cumlHandle_t handle,
148  double* input,
149  int n_rows,
150  int n_cols,
151  cumlSvmKernelType kernel,
152  int degree,
153  double gamma,
154  double coef0,
155  int n_support,
156  double b,
157  double* dual_coefs,
158  double* x_support,
159  int n_classes,
160  double* unique_labels,
161  double* preds,
162  double buffer_size,
163  int predict_class);
165 #ifdef __cplusplus
166 }
167 #endif
cuml_api.h
cumlHandle_t
int cumlHandle_t
Definition: cuml_api.h:26
cumlError_t
cumlError_t
Definition: cuml_api.h:28
cumlSpSvcPredict
cumlError_t cumlSpSvcPredict(cumlHandle_t handle, float *input, int n_rows, int n_cols, cumlSvmKernelType kernel, int degree, float gamma, float coef0, int n_support, float b, float *dual_coefs, float *x_support, int n_classes, float *unique_labels, float *preds, float buffer_size, int predict_class)
Definition: svm_api.cpp:164
cumlDpSvcPredict
cumlError_t cumlDpSvcPredict(cumlHandle_t handle, double *input, int n_rows, int n_cols, cumlSvmKernelType kernel, int degree, double gamma, double coef0, int n_support, double b, double *dual_coefs, double *x_support, int n_classes, double *unique_labels, double *preds, double buffer_size, int predict_class)
Definition: svm_api.cpp:219
cumlSpSvcFit
cumlError_t cumlSpSvcFit(cumlHandle_t handle, float *input, int n_rows, int n_cols, float *labels, float C, float cache_size, int max_iter, int nochange_steps, float tol, int verbosity, cumlSvmKernelType kernel, int degree, float gamma, float coef0, int *n_support, float *b, float **dual_coefs, float **x_support, int **support_idx, int *n_classes, float **unique_labels)
Definition: svm_api.cpp:16
cumlDpSvcFit
cumlError_t cumlDpSvcFit(cumlHandle_t handle, double *input, int n_rows, int n_cols, double *labels, double C, double cache_size, int max_iter, int nochange_steps, double tol, int verbosity, cumlSvmKernelType kernel, int degree, double gamma, double coef0, int *n_support, double *b, double **dual_coefs, double **x_support, int **support_idx, int *n_classes, double **unique_labels)
Definition: svm_api.cpp:90
ML::HoltWinters::buffer_size
void buffer_size(int n, int batch_size, int frequency, int *start_leveltrend_len, int *start_season_len, int *components_len, int *error_len, int *leveltrend_coef_shift, int *season_coef_shift)
cumlSvmKernelType
cumlSvmKernelType
Definition: svm_api.h:13
POLYNOMIAL
@ POLYNOMIAL
Definition: svm_api.h:13
TANH
@ TANH
Definition: svm_api.h:13
RBF
@ RBF
Definition: svm_api.h:13
LINEAR
@ LINEAR
Definition: svm_api.h:13
Generated by doxygen 1.9.1