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 | ||
) |