Classes | Namespaces | Functions
svc.hpp File Reference
#include "svm_model.h"
#include "svm_parameter.h"
#include <cuml/common/logger.hpp>
#include <raft/core/handle.hpp>
#include <raft/distance/distance_types.hpp>
Include dependency graph for svc.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

class  ML::SVM::SVC< math_t >
 C-Support Vector Classification. More...
 

Namespaces

 ML
 
 ML::SVM
 

Functions

template<typename math_t >
void ML::SVM::svcFit (const raft::handle_t &handle, math_t *input, int n_rows, int n_cols, math_t *labels, const SvmParameter &param, raft::distance::kernels::KernelParams &kernel_params, SvmModel< math_t > &model, const math_t *sample_weight)
 Fit a support vector classifier to the training data. More...
 
template<typename math_t >
void ML::SVM::svcFitSparse (const raft::handle_t &handle, int *indptr, int *indices, math_t *data, int n_rows, int n_cols, int nnz, math_t *labels, const SvmParameter &param, raft::distance::kernels::KernelParams &kernel_params, SvmModel< math_t > &model, const math_t *sample_weight)
 Fit a support vector classifier to the training data. More...
 
template<typename math_t >
void ML::SVM::svcPredict (const raft::handle_t &handle, math_t *input, int n_rows, int n_cols, raft::distance::kernels::KernelParams &kernel_params, const SvmModel< math_t > &model, math_t *preds, math_t buffer_size, bool predict_class)
 Predict classes or decision function value for samples in input. More...
 
template<typename math_t >
void ML::SVM::svcPredictSparse (const raft::handle_t &handle, int *indptr, int *indices, math_t *data, int n_rows, int n_cols, int nnz, raft::distance::kernels::KernelParams &kernel_params, const SvmModel< math_t > &model, math_t *preds, math_t buffer_size, bool predict_class)
 Predict classes or decision function value for samples in input. More...
 
template<typename math_t >
void ML::SVM::svmFreeBuffers (const raft::handle_t &handle, SvmModel< math_t > &m)