linear.hpp File Reference

#include <cuml/common/logger.hpp>
#include <raft/core/handle.hpp>

Include dependency graph for linear.hpp:

Go to the source code of this file.

Classes
struct	ML::SVM::linear::Params

Namespaces
	ML
	ML::SVM
	ML::SVM::linear

Functions
template<typename T >
int	ML::SVM::linear::fit (const raft::handle_t &handle, const Params ¶ms, const std::size_t nRows, const std::size_t nCols, const int nClasses, const T *classes, const T *X, const T *y, const T *sampleWeight, T *w, T *probScale)
	Fit a linear SVM model. More...
template<typename T >
void	ML::SVM::linear::computeProbabilities (const raft::handle_t &handle, const std::size_t nRows, const int nClasses, const T *probScale, T *scores, T *out)
	Compute probabilities from decision function scores. More...