#include "glm.hpp"#include <cumlprims/opg/matrix/data.hpp>#include <cumlprims/opg/matrix/part_descriptor.hpp>
Include dependency graph for ridge_mg.hpp:
Go to the source code of this file.
Namespaces | |
| ML | |
| ML::Ridge | |
| ML::Ridge::opg | |
Functions | |
| void | ML::Ridge::opg::fit (raft::handle_t &handle, std::vector< MLCommon::Matrix::Data< float > * > &input_data, MLCommon::Matrix::PartDescriptor &input_desc, std::vector< MLCommon::Matrix::Data< float > * > &labels, float *alpha, int n_alpha, float *coef, float *intercept, bool fit_intercept, bool normalize, int algo, bool verbose) |
| performs MNMG fit operation for the ridge regression More... | |
| void | ML::Ridge::opg::fit (raft::handle_t &handle, std::vector< MLCommon::Matrix::Data< double > * > &input_data, MLCommon::Matrix::PartDescriptor &input_desc, std::vector< MLCommon::Matrix::Data< double > * > &labels, double *alpha, int n_alpha, double *coef, double *intercept, bool fit_intercept, bool normalize, int algo, bool verbose) |
| void | ML::Ridge::opg::predict (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, size_t n_parts, MLCommon::Matrix::Data< float > **input, size_t n_rows, size_t n_cols, float *coef, float intercept, MLCommon::Matrix::Data< float > **preds, bool verbose) |
| performs MNMG prediction for OLS More... | |
| void | ML::Ridge::opg::predict (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, size_t n_parts, MLCommon::Matrix::Data< double > **input, size_t n_rows, size_t n_cols, double *coef, double intercept, MLCommon::Matrix::Data< double > **preds, bool verbose) |
