ML::OLS::opg Namespace Reference

Functions
void	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 algo, bool verbose)
	performs MNMG fit operation for the ridge regression More...
void	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 algo, bool verbose)
void	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	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)

Function Documentation

fit() [1/2]

void ML::OLS::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	algo,
		bool	verbose
	)

fit() [2/2]

void ML::OLS::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	algo,
		bool	verbose
	)

performs MNMG fit operation for the ridge regression

Parameters

[in]	handle	the internal cuml handle object
[in]	input_data	vector holding all partitions for that rank
[in]	input_desc	PartDescriptor object for the input
[in]	labels	labels data
[out]	coef	learned regression coefficients
[out]	intercept	intercept value
[in]	fit_intercept	fit intercept or not
[in]	normalize	normalize the data or not
[in]	algo	which algorithm is used for OLS. 0 is for SVD, 1 is for eig.
[in]	verbose

predict() [1/2]

void ML::OLS::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
	)

predict() [2/2]

void ML::OLS::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

Parameters

[in]	handle	the internal cuml handle object
[in]	rank_sizes	includes all the partition size information for the rank
[in]	n_parts	number of partitions
[in]	input	input data
[in]	n_rows	number of rows of input data
[in]	n_cols	number of cols of input data
[in]	coef	OLS coefficients
[in]	intercept	the fit intercept
[out]	preds	predictions
[in]	verbose