Class RegressionProblem

Class Documentation

class RegressionProblem

Public Functions

inline RegressionProblem(std::vector<std::vector<double>> input_data, std::vector<std::vector<double>> output_data)
inline virtual UINT get_input_dim() const
inline virtual std::vector<double> get_input_data(UINT sample_id) const
inline virtual UINT get_output_dim() const
inline virtual std::vector<double> get_output_data(UINT sample_id)
inline virtual double compute_loss(UINT sample_id, std::vector<double> prediction)

Protected Attributes

std::function<double(std::vector<double>, std::vector<double>)> _loss_function = loss_function::L2_distance<double>
std::vector<std::vector<double>> _input_data
std::vector<std::vector<double>> _output_data