Functions | |
| 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) |
regression model
| input | device pointer to feature matrix n_rows x n_cols |
| n_rows | number of rows of the feature matrix |
| n_cols | number of columns of the feature matrix |
| coef | weights of the model |
| intercept | bias term of the model |
| preds | device pointer to store predictions of size n_rows |
| 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 | ||
| ) |
| 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 | ||
| ) |