preprocess_mg.hpp File Reference

#include <cumlprims/opg/matrix/data.hpp>
#include <cumlprims/opg/matrix/part_descriptor.hpp>
#include <raft/core/comms.hpp>
#include <raft/core/handle.hpp>

Include dependency graph for preprocess_mg.hpp:

Go to the source code of this file.

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

Functions
void	ML::GLM::opg::preProcessData (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 *mu_input, float *mu_labels, float *norm2_input, bool fit_intercept, bool normalize, cudaStream_t *streams, int n_streams, bool verbose)
void	ML::GLM::opg::preProcessData (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 *mu_input, double *mu_labels, double *norm2_input, bool fit_intercept, bool normalize, cudaStream_t *streams, int n_streams, bool verbose)
void	ML::GLM::opg::postProcessData (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, float *mu_input, float *mu_labels, float *norm2_input, bool fit_intercept, bool normalize, cudaStream_t *streams, int n_streams, bool verbose)
void	ML::GLM::opg::postProcessData (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, double *mu_input, double *mu_labels, double *norm2_input, bool fit_intercept, bool normalize, cudaStream_t *streams, int n_streams, bool verbose)