|
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 ¶ms, 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 ¶ms, 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 ¶ms, 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 ¶ms, 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 ¶ms, 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 ¶ms, 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...
|
|