pca.hpp
Go to the documentation of this file.
1 /*
2  * SPDX-FileCopyrightText: Copyright (c) 2018-2025, NVIDIA CORPORATION.
3  * SPDX-License-Identifier: Apache-2.0
4  */
5 
6 #pragma once
7 
8 #include "params.hpp"
9 
10 namespace raft {
11 class handle_t;
12 }
13 
14 namespace ML {
15 
16 void pcaFit(raft::handle_t& handle,
17  float* input,
18  float* components,
19  float* explained_var,
20  float* explained_var_ratio,
21  float* singular_vals,
22  float* mu,
23  float* noise_vars,
24  const paramsPCA& prms,
25  bool flip_signs_based_on_U);
26 void pcaFit(raft::handle_t& handle,
27  double* input,
28  double* components,
29  double* explained_var,
30  double* explained_var_ratio,
31  double* singular_vals,
32  double* mu,
33  double* noise_vars,
34  const paramsPCA& prms,
35  bool flip_signs_based_on_U);
36 void pcaFitTransform(raft::handle_t& handle,
37  float* input,
38  float* trans_input,
39  float* components,
40  float* explained_var,
41  float* explained_var_ratio,
42  float* singular_vals,
43  float* mu,
44  float* noise_vars,
45  const paramsPCA& prms,
46  bool flip_signs_based_on_U);
47 void pcaFitTransform(raft::handle_t& handle,
48  double* input,
49  double* trans_input,
50  double* components,
51  double* explained_var,
52  double* explained_var_ratio,
53  double* singular_vals,
54  double* mu,
55  double* noise_vars,
56  const paramsPCA& prms,
57  bool flip_signs_based_on_U);
58 void pcaInverseTransform(raft::handle_t& handle,
59  float* trans_input,
60  float* components,
61  float* singular_vals,
62  float* mu,
63  float* input,
64  const paramsPCA& prms);
65 void pcaInverseTransform(raft::handle_t& handle,
66  double* trans_input,
67  double* components,
68  double* singular_vals,
69  double* mu,
70  double* input,
71  const paramsPCA& prms);
72 void pcaTransform(raft::handle_t& handle,
73  float* input,
74  float* components,
75  float* trans_input,
76  float* singular_vals,
77  float* mu,
78  const paramsPCA& prms);
79 void pcaTransform(raft::handle_t& handle,
80  double* input,
81  double* components,
82  double* trans_input,
83  double* singular_vals,
84  double* mu,
85  const paramsPCA& prms);
86 
87 }; // end namespace ML
ML::paramsPCATemplate
structure for pca parameters. Ref: http://scikit-learn.org/stable/modules/generated/sklearn....
Definition: params.hpp:64
ML
Definition: dbscan.hpp:18
ML::pcaFit
void pcaFit(raft::handle_t &handle, float *input, float *components, float *explained_var, float *explained_var_ratio, float *singular_vals, float *mu, float *noise_vars, const paramsPCA &prms, bool flip_signs_based_on_U)
ML::pcaTransform
void pcaTransform(raft::handle_t &handle, float *input, float *components, float *trans_input, float *singular_vals, float *mu, const paramsPCA &prms)
ML::pcaInverseTransform
void pcaInverseTransform(raft::handle_t &handle, float *trans_input, float *components, float *singular_vals, float *mu, float *input, const paramsPCA &prms)
ML::pcaFitTransform
void pcaFitTransform(raft::handle_t &handle, float *input, float *trans_input, float *components, float *explained_var, float *explained_var_ratio, float *singular_vals, float *mu, float *noise_vars, const paramsPCA &prms, bool flip_signs_based_on_U)
raft
Definition: dbscan.hpp:14
Generated by doxygen 1.9.1