Commit bf2ee3c5 authored by Rahul Kavi's avatar Rahul Kavi Committed by Maksim Shabunin

updated prototype for logistic regression classifier

parent 3a6deb3e
...@@ -581,17 +581,17 @@ struct CV_EXPORTS LogisticRegressionParams ...@@ -581,17 +581,17 @@ struct CV_EXPORTS LogisticRegressionParams
int regularized; int regularized;
int train_method; int train_method;
int mini_batch_size; int mini_batch_size;
CvTermCriteria term_crit; cv::TermCriteria term_crit;
LogisticRegressionParams(); 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 class CV_EXPORTS LogisticRegression
{ {
public: public:
LogisticRegression( const LogisticRegressionParams& params);
LogisticRegression();
LogisticRegression(cv::InputArray data_ip, cv::InputArray labels_ip, const LogisticRegressionParams& params); LogisticRegression(cv::InputArray data_ip, cv::InputArray labels_ip, const LogisticRegressionParams& params);
virtual ~LogisticRegression(); virtual ~LogisticRegression();
...@@ -601,10 +601,11 @@ public: ...@@ -601,10 +601,11 @@ public:
virtual bool train(cv::InputArray data_ip, cv::InputArray label_ip); virtual bool train(cv::InputArray data_ip, cv::InputArray label_ip);
virtual void predict( cv::InputArray data, cv::OutputArray predicted_labels ) const; virtual void predict( cv::InputArray data, cv::OutputArray predicted_labels ) const;
virtual void save(std::string filepath) const; virtual void write(FileStorage& fs) const;
virtual void load(const std::string filepath); virtual void read(const FileNode& fn);
cv::Mat get_learnt_thetas() const; const cv::Mat get_learnt_thetas() const;
virtual void clear();
protected: protected:
...@@ -623,11 +624,6 @@ 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 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); virtual bool set_label_map(const cv::Mat& labels);
static cv::Mat remap_labels(const cv::Mat& labels, const std::map<int, int>& lmap); 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 }// namespace cv
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment