linear_regression.h

#ifndef LINEAR_REGRESSION_H
#define LINEAR_REGRESSION_H
 
#include "PlusML/gradient.h"
#include "PlusML/util.h"
 
#include <cstdint>
#include <Eigen/Dense>
 
namespace plusml {
class EXPORT LinearRegression {
public:
  LinearRegression() = delete;
 
  explicit LinearRegression(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,
              const LossGradient& grad,
              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 //LINEAR_REGRESSION_H