qn_mg.hpp File Reference

#include <cuda_runtime.h>
#include <cuml/common/logger.hpp>
#include <cuml/linear_model/qn.h>
#include <raft/core/comms.hpp>
#include <cumlprims/opg/matrix/data.hpp>
#include <cumlprims/opg/matrix/part_descriptor.hpp>
#include <vector>

Include dependency graph for qn_mg.hpp:

Go to the source code of this file.

Namespaces
	ML
	ML::GLM
	ML::GLM::opg

Functions
std::vector< float >	ML::GLM::opg::getUniquelabelsMG (const raft::handle_t &handle, Matrix::PartDescriptor &input_desc, std::vector< Matrix::Data< float > * > &labels)
	Calculate unique class labels across multiple GPUs in a multi-node environment. More...
void	ML::GLM::opg::qnFit (raft::handle_t &handle, std::vector< Matrix::Data< float > * > &input_data, Matrix::PartDescriptor &input_desc, std::vector< Matrix::Data< float > * > &labels, float *coef, const qn_params &pams, bool X_col_major, int n_classes, float *f, int *num_iters)
	performs MNMG fit operation for the logistic regression using quasi newton methods More...
void	ML::GLM::opg::qnFitSparse (raft::handle_t &handle, std::vector< Matrix::Data< float > * > &input_values, int *input_cols, int *input_row_ids, int X_nnz, Matrix::PartDescriptor &input_desc, std::vector< Matrix::Data< float > * > &labels, float *coef, const qn_params &pams, int n_classes, float *f, int *num_iters)
	support sparse vectors (Compressed Sparse Row format) for MNMG logistic regression fit using quasi newton methods More...