diff --git a/modules/ml/include/opencv2/ml.hpp b/modules/ml/include/opencv2/ml.hpp index 9575901c6bf1fcf2985d0663caec2b918519a1dc..aa3df06ac9b34dfbbff090db45828161470b902e 100644 --- a/modules/ml/include/opencv2/ml.hpp +++ b/modules/ml/include/opencv2/ml.hpp @@ -609,12 +609,14 @@ public: protected: LogisticRegressionParams params; - cv::Mat learnt_thetas; + cv::Mat learnt_thetas; std::string default_model_name; std::map forward_mapper; std::map reverse_mapper; + cv::Mat labels_o; cv::Mat labels_n; + static cv::Mat calc_sigmoid(const cv::Mat& data); virtual double compute_cost(const cv::Mat& data, const cv::Mat& labels, const cv::Mat& init_theta); virtual cv::Mat compute_batch_gradient(const cv::Mat& data, const cv::Mat& labels, const cv::Mat& init_theta);