ML::Internals::GraphBasedDimRedCallback Class Reference

#include <callback.hpp>

Inheritance diagram for ML::Internals::GraphBasedDimRedCallback:

Collaboration diagram for ML::Internals::GraphBasedDimRedCallback:

Public Member Functions
template<typename T >
void	setup (int n, int n_components)
virtual void	on_preprocess_end (void *embeddings)=0
virtual void	on_epoch_end (void *embeddings)=0
virtual void	on_train_end (void *embeddings)=0
Public Member Functions inherited from ML::Internals::Callback
virtual	~Callback ()

Protected Attributes
int	n
int	n_components
bool	isFloat

Member Function Documentation

on_epoch_end()

virtual void ML::Internals::GraphBasedDimRedCallback::on_epoch_end ( void *  embeddings )

pure virtual

on_preprocess_end()

virtual void ML::Internals::GraphBasedDimRedCallback::on_preprocess_end ( void *  embeddings )

pure virtual

on_train_end()

virtual void ML::Internals::GraphBasedDimRedCallback::on_train_end ( void *  embeddings )

pure virtual

setup()

template<typename T >

void ML::Internals::GraphBasedDimRedCallback::setup ( int  n, int  n_components  )

inline

Member Data Documentation

isFloat

bool ML::Internals::GraphBasedDimRedCallback::isFloat

protected

n

int ML::Internals::GraphBasedDimRedCallback::n

protected

n_components

int ML::Internals::GraphBasedDimRedCallback::n_components

protected

---

The documentation for this class was generated from the following file:

• include/cuml/common/callback.hpp