diff options
author | Anton Samokhvalov <pg83@yandex.ru> | 2022-02-10 16:45:15 +0300 |
---|---|---|
committer | Daniil Cherednik <dcherednik@yandex-team.ru> | 2022-02-10 16:45:15 +0300 |
commit | 72cb13b4aff9bc9cf22e49251bc8fd143f82538f (patch) | |
tree | da2c34829458c7d4e74bdfbdf85dff449e9e7fb8 /library/cpp/linear_regression/linear_model.h | |
parent | 778e51ba091dc39e7b7fcab2b9cf4dbedfb6f2b5 (diff) | |
download | ydb-72cb13b4aff9bc9cf22e49251bc8fd143f82538f.tar.gz |
Restoring authorship annotation for Anton Samokhvalov <pg83@yandex.ru>. Commit 1 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 8bb050cff7..551bfe36dc 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); + } }; |