#include <cuml/linear_model/glm.hpp>#include <cumlprims/opg/matrix/data.hpp>#include <cumlprims/opg/matrix/part_descriptor.hpp>
Include dependency graph for cd_mg.hpp:
Go to the source code of this file.
Namespaces | |
| ML | |
| ML::CD | |
| ML::CD::opg | |
Functions | |
| void | ML::CD::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 *coef, float *intercept, bool fit_intercept, bool normalize, int epochs, float alpha, float l1_ratio, bool shuffle, float tol, bool verbose) |
| performs MNMG fit operation for the ridge regression More... | |
| void | ML::CD::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 *coef, double *intercept, bool fit_intercept, bool normalize, int epochs, double alpha, double l1_ratio, bool shuffle, double tol, bool verbose) |
| void | ML::CD::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::CD::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) |
