cuML C++ API: include/cuml/solvers/solver.hpp File Reference

Go to the source code of this file.

Namespaces
	raft

	ML

	ML::Solver

Functions
void	ML::Solver::sgdFit (raft::handle_t &handle, float input, int n_rows, int n_cols, float labels, float coef, float intercept, bool fit_intercept, int batch_size, int epochs, int lr_type, float eta0, float power_t, int loss, int penalty, float alpha, float l1_ratio, bool shuffle, float tol, int n_iter_no_change)

void	ML::Solver::sgdFit (raft::handle_t &handle, double input, int n_rows, int n_cols, double labels, double coef, double intercept, bool fit_intercept, int batch_size, int epochs, int lr_type, double eta0, double power_t, int loss, int penalty, double alpha, double l1_ratio, bool shuffle, double tol, int n_iter_no_change)

void	ML::Solver::sgdPredict (raft::handle_t &handle, const float input, int n_rows, int n_cols, const float coef, float intercept, float *preds, int loss)

void	ML::Solver::sgdPredict (raft::handle_t &handle, const double input, int n_rows, int n_cols, const double coef, double intercept, double *preds, int loss)

void	ML::Solver::sgdPredictBinaryClass (raft::handle_t &handle, const float input, int n_rows, int n_cols, const float coef, float intercept, float *preds, int loss)

void	ML::Solver::sgdPredictBinaryClass (raft::handle_t &handle, const double input, int n_rows, int n_cols, const double coef, double intercept, double *preds, int loss)

void	ML::Solver::cdFit (raft::handle_t &handle, float input, int n_rows, int n_cols, float labels, float coef, float intercept, bool fit_intercept, bool normalize, int epochs, int loss, float alpha, float l1_ratio, bool shuffle, float tol, float *sample_weight=nullptr)

void	ML::Solver::cdFit (raft::handle_t &handle, double input, int n_rows, int n_cols, double labels, double coef, double intercept, bool fit_intercept, bool normalize, int epochs, int loss, double alpha, double l1_ratio, bool shuffle, double tol, double *sample_weight=nullptr)

void	ML::Solver::cdPredict (raft::handle_t &handle, const float input, int n_rows, int n_cols, const float coef, float intercept, float *preds, int loss)

void	ML::Solver::cdPredict (raft::handle_t &handle, const double input, int n_rows, int n_cols, const double coef, double intercept, double *preds, int loss)