Functions
to make predictions with a fitted ordinary least squares and ridge

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)
 

Detailed Description

regression model

Parameters
inputdevice pointer to feature matrix n_rows x n_cols
n_rowsnumber of rows of the feature matrix
n_colsnumber of columns of the feature matrix
coefweights of the model
interceptbias term of the model
predsdevice pointer to store predictions of size n_rows

Function Documentation

◆ gemmPredict() [1/2]

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 
)

◆ gemmPredict() [2/2]

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 
)