Namespaces | Functions
glm.hpp File Reference
#include <cuml/linear_model/qn.h>
#include <raft/core/handle.hpp>
Include dependency graph for glm.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

 ML
 
 ML::GLM
 

Functions

void ML::GLM::olsFit (const raft::handle_t &handle, float *input, size_t n_rows, size_t n_cols, float *labels, float *coef, float *intercept, bool fit_intercept, bool normalize, int algo=0, float *sample_weight=nullptr)
 
void ML::GLM::olsFit (const raft::handle_t &handle, double *input, size_t n_rows, size_t n_cols, double *labels, double *coef, double *intercept, bool fit_intercept, bool normalize, int algo=0, double *sample_weight=nullptr)
 
void ML::GLM::ridgeFit (const raft::handle_t &handle, float *input, size_t n_rows, size_t n_cols, float *labels, float *alpha, int n_alpha, float *coef, float *intercept, bool fit_intercept, bool normalize, int algo=0, float *sample_weight=nullptr)
 
void ML::GLM::ridgeFit (const raft::handle_t &handle, double *input, size_t n_rows, size_t n_cols, double *labels, double *alpha, int n_alpha, double *coef, double *intercept, bool fit_intercept, bool normalize, int algo=0, double *sample_weight=nullptr)
 
void ML::GLM::gemmPredict (const raft::handle_t &handle, const float *input, size_t n_rows, size_t n_cols, const float *coef, float intercept, float *preds)
 
void ML::GLM::gemmPredict (const raft::handle_t &handle, const double *input, size_t n_rows, size_t n_cols, const double *coef, double intercept, double *preds)
 
template<typename T , typename I = int>
void ML::GLM::qnFit (const raft::handle_t &cuml_handle, const qn_params &params, T *X, bool X_col_major, T *y, I N, I D, I C, T *w0, T *f, int *num_iters, T *sample_weight=nullptr, T svr_eps=0)
 Fit a GLM using quasi newton methods. More...
 
template<typename T , typename I = int>
void ML::GLM::qnFitSparse (const raft::handle_t &cuml_handle, const qn_params &params, T *X_values, I *X_cols, I *X_row_ids, I X_nnz, T *y, I N, I D, I C, T *w0, T *f, int *num_iters, T *sample_weight=nullptr, T svr_eps=0)
 Fit a GLM using quasi newton methods. More...
 
template<typename T , typename I = int>
void ML::GLM::qnDecisionFunction (const raft::handle_t &cuml_handle, const qn_params &params, T *X, bool X_col_major, I N, I D, I C, T *coefs, T *scores)
 Obtain the confidence scores of samples. More...
 
template<typename T , typename I = int>
void ML::GLM::qnDecisionFunctionSparse (const raft::handle_t &cuml_handle, const qn_params &params, T *X_values, I *X_cols, I *X_row_ids, I X_nnz, I N, I D, I C, T *coefs, T *scores)
 Obtain the confidence scores of samples. More...
 
template<typename T , typename I = int>
void ML::GLM::qnPredict (const raft::handle_t &cuml_handle, const qn_params &params, T *X, bool X_col_major, I N, I D, I C, T *coefs, T *preds)
 Predict a GLM using quasi newton methods. More...
 
template<typename T , typename I = int>
void ML::GLM::qnPredictSparse (const raft::handle_t &cuml_handle, const qn_params &params, T *X_values, I *X_cols, I *X_row_ids, I X_nnz, I N, I D, I C, T *coefs, T *preds)
 Predict a GLM using quasi newton methods. More...