#include "tsvd.hpp"#include <cumlprims/opg/matrix/data.hpp>#include <cumlprims/opg/matrix/part_descriptor.hpp>
Include dependency graph for tsvd_mg.hpp:
Go to the source code of this file.
Namespaces | |
| ML | |
| ML::TSVD | |
| ML::TSVD::opg | |
Functions | |
| void | ML::TSVD::opg::fit (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, std::uint32_t n_parts, MLCommon::Matrix::floatData_t **input, float *components, float *singular_vals, paramsTSVDMG &prms, bool verbose=false) |
| performs MNMG fit operation for the tsvd More... | |
| void | ML::TSVD::opg::fit (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, std::uint32_t n_parts, MLCommon::Matrix::doubleData_t **input, double *components, double *singular_vals, paramsTSVDMG &prms, bool verbose=false) |
| void | ML::TSVD::opg::fit_transform (raft::handle_t &handle, std::vector< MLCommon::Matrix::Data< float > * > &input_data, MLCommon::Matrix::PartDescriptor &input_desc, std::vector< MLCommon::Matrix::Data< float > * > &trans_data, MLCommon::Matrix::PartDescriptor &trans_desc, float *components, float *explained_var, float *explained_var_ratio, float *singular_vals, paramsTSVDMG &prms, bool verbose) |
| performs MNMG fit and transform operation for the tsvd. More... | |
| void | ML::TSVD::opg::fit_transform (raft::handle_t &handle, std::vector< MLCommon::Matrix::Data< double > * > &input_data, MLCommon::Matrix::PartDescriptor &input_desc, std::vector< MLCommon::Matrix::Data< double > * > &trans_data, MLCommon::Matrix::PartDescriptor &trans_desc, double *components, double *explained_var, double *explained_var_ratio, double *singular_vals, paramsTSVDMG &prms, bool verbose) |
| void | ML::TSVD::opg::transform (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, std::uint32_t n_parts, MLCommon::Matrix::Data< float > **input, float *components, MLCommon::Matrix::Data< float > **trans_input, paramsTSVDMG &prms, bool verbose) |
| performs MNMG transform operation for the tsvd. More... | |
| void | ML::TSVD::opg::transform (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, std::uint32_t n_parts, MLCommon::Matrix::Data< double > **input, double *components, MLCommon::Matrix::Data< double > **trans_input, paramsTSVDMG &prms, bool verbose) |
| void | ML::TSVD::opg::inverse_transform (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, std::uint32_t n_parts, MLCommon::Matrix::Data< float > **trans_input, float *components, MLCommon::Matrix::Data< float > **input, paramsTSVDMG &prms, bool verbose) |
| performs MNMG inverse transform operation for the output. More... | |
| void | ML::TSVD::opg::inverse_transform (raft::handle_t &handle, MLCommon::Matrix::RankSizePair **rank_sizes, std::uint32_t n_parts, MLCommon::Matrix::Data< double > **trans_input, double *components, MLCommon::Matrix::Data< double > **input, paramsTSVDMG &prms, bool verbose) |
