diff options
author | Anton Samokhvalov <pg83@yandex.ru> | 2022-02-10 16:45:17 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:45:17 +0300 |
commit | d3a398281c6fd1d3672036cb2d63f842d2cb28c5 (patch) | |
tree | dd4bd3ca0f36b817e96812825ffaf10d645803f2 /library/cpp/linear_regression/linear_model.h | |
parent | 72cb13b4aff9bc9cf22e49251bc8fd143f82538f (diff) | |
download | ydb-d3a398281c6fd1d3672036cb2d63f842d2cb28c5.tar.gz |
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/linear_regression/linear_model.h')
-rw-r--r-- | library/cpp/linear_regression/linear_model.h | 62 |
1 files changed, 31 insertions, 31 deletions
diff --git a/library/cpp/linear_regression/linear_model.h b/library/cpp/linear_regression/linear_model.h index 551bfe36dc..8bb050cff7 100644 --- a/library/cpp/linear_regression/linear_model.h +++ b/library/cpp/linear_regression/linear_model.h @@ -8,35 +8,35 @@ #include <utility> class TLinearModel { -private: - TVector<double> Coefficients; - double Intercept; - -public: - Y_SAVELOAD_DEFINE(Coefficients, Intercept); - - TLinearModel(TVector<double>&& coefficients, const double intercept) - : Coefficients(std::move(coefficients)) - , Intercept(intercept) - { - } - - explicit TLinearModel(size_t featuresCount = 0) - : Coefficients(featuresCount) - , Intercept(0.) - { - } - - const TVector<double>& GetCoefficients() const { - return Coefficients; - } - - double GetIntercept() const { - return Intercept; - } - - template <typename T> - double Prediction(const TVector<T>& features) const { - return InnerProduct(Coefficients, features, Intercept); - } +private: + TVector<double> Coefficients; + double Intercept; + +public: + Y_SAVELOAD_DEFINE(Coefficients, Intercept); + + TLinearModel(TVector<double>&& coefficients, const double intercept) + : Coefficients(std::move(coefficients)) + , Intercept(intercept) + { + } + + explicit TLinearModel(size_t featuresCount = 0) + : Coefficients(featuresCount) + , Intercept(0.) + { + } + + const TVector<double>& GetCoefficients() const { + return Coefficients; + } + + double GetIntercept() const { + return Intercept; + } + + template <typename T> + double Prediction(const TVector<T>& features) const { + return InnerProduct(Coefficients, features, Intercept); + } }; |