binary_logistic_regression.h

#ifndef BINARY_LOGISTIC_REGRESSION_H
#define BINARY_LOGISTIC_REGRESSION_H
 
#include "PlusML/util.h"
#include <cstdint>
#include <Eigen/Dense>
 
namespace plusml {
class EXPORT BinaryLogisticRegression {
public:
  BinaryLogisticRegression() = delete;
 
  explicit BinaryLogisticRegression(uint64_t features, bool bias_enabled = true);
 
  uint64_t Features() const;
 
  Eigen::MatrixXf Parameters() const;
 
  void SetParameters(const Eigen::MatrixXf& parameters);
 
  void FitSGD(const Eigen::MatrixXf& samples,
              const Eigen::MatrixXf& targets,
              float learning_rate,
              uint64_t batch_size,
              uint64_t epochs);
 
  Eigen::MatrixXf Predict(Eigen::MatrixXf samples);
 
private:
  uint64_t features_;
  bool bias_enabled_;
  Eigen::MatrixXf w_;
};
} //namespace plusml
 
#endif //BINARY_LOGISTIC_REGRESSION_H