preprocess_mg.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2020-2022, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 
6 #pragma once
7 
8 #include <cumlprims/opg/matrix/data.hpp>
9 #include <cumlprims/opg/matrix/part_descriptor.hpp>
10 #include <raft/core/comms.hpp>
11 #include <raft/core/handle.hpp>
12 
13 namespace ML {
14 namespace GLM {
15 namespace opg {
16 
17 void preProcessData(raft::handle_t& handle,
18  std::vector<MLCommon::Matrix::Data<float>*>& input_data,
19  MLCommon::Matrix::PartDescriptor& input_desc,
20  std::vector<MLCommon::Matrix::Data<float>*>& labels,
21  float* mu_input,
22  float* mu_labels,
23  float* norm2_input,
24  bool fit_intercept,
25  bool normalize,
26  cudaStream_t* streams,
27  int n_streams,
28  bool verbose);
29 
30 void preProcessData(raft::handle_t& handle,
31  std::vector<MLCommon::Matrix::Data<double>*>& input_data,
32  MLCommon::Matrix::PartDescriptor& input_desc,
33  std::vector<MLCommon::Matrix::Data<double>*>& labels,
34  double* mu_input,
35  double* mu_labels,
36  double* norm2_input,
37  bool fit_intercept,
38  bool normalize,
39  cudaStream_t* streams,
40  int n_streams,
41  bool verbose);
42 
43 void postProcessData(raft::handle_t& handle,
44  std::vector<MLCommon::Matrix::Data<float>*>& input_data,
45  MLCommon::Matrix::PartDescriptor& input_desc,
46  std::vector<MLCommon::Matrix::Data<float>*>& labels,
47  float* coef,
48  float* intercept,
49  float* mu_input,
50  float* mu_labels,
51  float* norm2_input,
52  bool fit_intercept,
53  bool normalize,
54  cudaStream_t* streams,
55  int n_streams,
56  bool verbose);
57 
58 void postProcessData(raft::handle_t& handle,
59  std::vector<MLCommon::Matrix::Data<double>*>& input_data,
60  MLCommon::Matrix::PartDescriptor& input_desc,
61  std::vector<MLCommon::Matrix::Data<double>*>& labels,
62  double* coef,
63  double* intercept,
64  double* mu_input,
65  double* mu_labels,
66  double* norm2_input,
67  bool fit_intercept,
68  bool normalize,
69  cudaStream_t* streams,
70  int n_streams,
71  bool verbose);
72 
73 }; // end namespace opg
74 }; // namespace GLM
75 }; // end namespace ML
ML::GLM::opg::postProcessData
void 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)
ML::GLM::opg::preProcessData
void 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)
ML::HDBSCAN::detail::Utils::normalize
void normalize(value_t *data, value_idx n, size_t m, cudaStream_t stream)
Definition: utils.h:177
ML
Definition: dbscan.hpp:18
Generated by doxygen 1.9.1