svr.hpp File Reference

#include <cuml/matrix/kernelparams.h>
#include <cublas_v2.h>

Include dependency graph for svr.hpp:

Go to the source code of this file.

Namespaces
	ML
	ML::SVM

Functions
template<typename math_t >
void	ML::SVM::svrFit (const raft::handle_t &handle, math_t *X, int n_rows, int n_cols, math_t *y, const SvmParameter &param, MLCommon::Matrix::KernelParams &kernel_params, SvmModel< math_t > &model, const math_t *sample_weight=nullptr)
	Fit a support vector regressor to the training data. More...
template<typename math_t >
void	ML::SVM::svrFitSparse (const raft::handle_t &handle, int *indptr, int *indices, math_t *data, int n_rows, int n_cols, int nnz, math_t *y, const SvmParameter &param, raft::distance::kernels::KernelParams &kernel_params, SvmModel< math_t > &model, const math_t *sample_weight=nullptr)
	Fit a support vector regressor to the training data. More...