intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 42 insertions, 0 deletions

diff --git a/library/cpp/linear_regression/linear_model.h b/library/cpp/linear_regression/linear_model.h
new file mode 100644
index 0000000000..8bb050cff7
--- /dev/null
+++ b/library/cpp/linear_regression/linear_model.h
@@ -0,0 +1,42 @@
+#pragma once
+
+#include <util/generic/algorithm.h>
+#include <util/generic/vector.h>
+
+#include <util/ysaveload.h>
+
+#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);
+    }
+};