mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-01-18 22:44:02 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

updated prototype for logistic regression classifier

Browse Source
This commit is contained in:
Rahul Kavi 2013-11-05 05:33:14 -05:00 committed by Maksim Shabunin
parent 3a6deb3ed3
commit bf2ee3c58a
1 changed files with 8 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
modules/ml/include/opencv2/ml.hpp
View File

@ -581,17 +581,17 @@ struct CV_EXPORTS LogisticRegressionParams
int regularized;
int train_method;
int mini_batch_size;
CvTermCriteria term_crit;
cv::TermCriteria term_crit;
LogisticRegressionParams();
LogisticRegressionParams(double alpha, int num_iters, int norm, int regularized, int train_method, int minbatchsize);
LogisticRegressionParams(double learning_rate, int iters, int train_method, int normlization, int reg, int mini_batch_size);
};
class CV_EXPORTS LogisticRegression
{
public:
LogisticRegression();
LogisticRegression( const LogisticRegressionParams& params);
LogisticRegression(cv::InputArray data_ip, cv::InputArray labels_ip, const LogisticRegressionParams& params);
virtual ~LogisticRegression();
@ -601,10 +601,11 @@ public:
virtual bool train(cv::InputArray data_ip, cv::InputArray label_ip);
virtual void predict( cv::InputArray data, cv::OutputArray predicted_labels ) const;
virtual void save(std::string filepath) const;
virtual void load(const std::string filepath);
virtual void write(FileStorage& fs) const;
virtual void read(const FileNode& fn);
cv::Mat get_learnt_thetas() const;
const cv::Mat get_learnt_thetas() const;
virtual void clear();
protected:
@ -623,11 +624,6 @@ protected:
virtual cv::Mat compute_mini_batch_gradient(const cv::Mat& data, const cv::Mat& labels, const cv::Mat& init_theta);
virtual bool set_label_map(const cv::Mat& labels);
static cv::Mat remap_labels(const cv::Mat& labels, const std::map<int, int>& lmap);
virtual void write(FileStorage& fs) const;
virtual void read(const FileNode& fn);
virtual void clear();
};
}// namespace cv