ML::CD::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 epochs, float alpha, float l1_ratio, bool shuffle, float tol, 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 epochs, double alpha, double l1_ratio, bool shuffle, double tol, 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::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
	)

fit() [2/2]

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

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]	epochs	number of epochs
[in]	alpha	ridge parameter
[in]	l1_ratio	l1 ratio
[in]	shuffle	whether to shuffle the data
[in]	tol	tolerance for early stopping during fitting
[in]	verbose

predict() [1/2]

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
	)

predict() [2/2]

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

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