Namespaces | Functions
make_regression.hpp File Reference
#include <cstdint>
Include dependency graph for make_regression.hpp:

Go to the source code of this file.

Namespaces

 raft
 
 ML
 
 ML::Datasets
 

Functions

void ML::Datasets::make_regression (const raft::handle_t &handle, float *out, float *values, int64_t n_rows, int64_t n_cols, int64_t n_informative, float *coef=nullptr, int64_t n_targets=1LL, float bias=0.0f, int64_t effective_rank=-1LL, float tail_strength=0.5f, float noise=0.0f, bool shuffle=true, uint64_t seed=0ULL)
 GPU-equivalent of sklearn.datasets.make_regression as documented at: https://scikit-learn.org/stable/modules/generated/sklearn.datasets.make_regression.html. More...
 
void ML::Datasets::make_regression (const raft::handle_t &handle, double *out, double *values, int64_t n_rows, int64_t n_cols, int64_t n_informative, double *coef=nullptr, int64_t n_targets=1LL, double bias=0.0, int64_t effective_rank=-1LL, double tail_strength=0.5, double noise=0.0, bool shuffle=true, uint64_t seed=0ULL)
 
void ML::Datasets::make_regression (const raft::handle_t &handle, float *out, float *values, int n_rows, int n_cols, int n_informative, float *coef=nullptr, int n_targets=1LL, float bias=0.0f, int effective_rank=-1LL, float tail_strength=0.5f, float noise=0.0f, bool shuffle=true, uint64_t seed=0ULL)
 
void ML::Datasets::make_regression (const raft::handle_t &handle, double *out, double *values, int n_rows, int n_cols, int n_informative, double *coef=nullptr, int n_targets=1LL, double bias=0.0, int effective_rank=-1LL, double tail_strength=0.5, double noise=0.0, bool shuffle=true, uint64_t seed=0ULL)