diff --git a/cpp/include/cuml/linear_model/glm.hpp b/cpp/include/cuml/linear_model/glm.hpp index d452aff3d1..19c3d734a8 100644 --- a/cpp/include/cuml/linear_model/glm.hpp +++ b/cpp/include/cuml/linear_model/glm.hpp @@ -27,7 +27,7 @@ namespace GLM { * @param n_cols number of columns of the feature matrix * @param labels device pointer to label vector of length n_rows * @param coef device pointer to hold the solution for weights of size n_cols - * @param intercept device pointer to hold the solution for bias term of size 1 + * @param intercept host pointer to hold the solution for bias term of size 1 * @param fit_intercept if true, fit intercept * @param normalize if true, normalize data to zero mean, unit variance * @param algo specifies which solver to use (0: SVD, 1: Eigendecomposition, 2: @@ -62,10 +62,10 @@ void olsFit(const raft::handle_t& handle, * @param n_rows number of rows of the feature matrix * @param n_cols number of columns of the feature matrix * @param labels device pointer to label vector of length n_rows - * @param alpha device pointer to parameters of the l2 regularizer + * @param alpha host pointer to parameters of the l2 regularizer * @param n_alpha number of regularization parameters * @param coef device pointer to hold the solution for weights of size n_cols - * @param intercept device pointer to hold the solution for bias term of size 1 + * @param intercept host pointer to hold the solution for bias term of size 1 * @param fit_intercept if true, fit intercept * @param normalize if true, normalize data to zero mean, unit variance * @param algo specifies which solver to use (0: SVD, 1: Eigendecomposition) diff --git a/cpp/src/glm/ols.cuh b/cpp/src/glm/ols.cuh index 9e2e8212e0..38b044e0ed 100644 --- a/cpp/src/glm/ols.cuh +++ b/cpp/src/glm/ols.cuh @@ -43,7 +43,7 @@ using namespace MLCommon; * @param n_cols number of columns of the feature matrix * @param labels device pointer to label vector of length n_rows * @param coef device pointer to hold the solution for weights of size n_cols - * @param intercept device pointer to hold the solution for bias term of size 1 + * @param intercept host pointer to hold the solution for bias term of size 1 * @param fit_intercept if true, fit intercept * @param normalize if true, normalize data to zero mean, unit variance * @param stream cuda stream diff --git a/cpp/src/glm/ridge.cuh b/cpp/src/glm/ridge.cuh index 6431eb0297..98e206344d 100644 --- a/cpp/src/glm/ridge.cuh +++ b/cpp/src/glm/ridge.cuh @@ -139,10 +139,10 @@ void ridgeEig(const raft::handle_t& handle, * @param n_rows number of rows of the feature matrix * @param n_cols number of columns of the feature matrix * @param labels device pointer to label vector of length n_rows - * @param alpha device pointer to parameters of the l2 regularizer + * @param alpha host pointer to parameters of the l2 regularizer * @param n_alpha number of regularization parameters * @param coef device pointer to hold the solution for weights of size n_cols - * @param intercept device pointer to hold the solution for bias term of size 1 + * @param intercept host pointer to hold the solution for bias term of size 1 * @param fit_intercept if true, fit intercept * @param normalize if true, normalize data to zero mean, unit variance * @param stream cuda stream