Class ClassificationProblem

Class Documentation

class ClassificationProblem

Public Functions

inline ClassificationProblem(std::vector<std::vector<double>> input_data, std::vector<UINT> label_data)
inline virtual UINT get_input_dim() const
inline virtual std::vector<double> get_input_data(UINT sample_id) const
inline virtual UINT get_category_count() const
inline virtual UINT get_output_data(UINT sample_id) const
inline virtual double compute_loss(UINT sample_id, std::vector<double> probability_distribution) const