summaryrefslogtreecommitdiffstats
path: root/library/cpp/linear_regression/linear_model.h
diff options
context:
space:
mode:
authoralex-sh <[email protected]>2022-02-10 16:50:03 +0300
committerDaniil Cherednik <[email protected]>2022-02-10 16:50:03 +0300
commit88ee78b1a163eaddee7e880ac73943456040fce0 (patch)
tree5d5cb817648f650d76cf1076100726fd9b8448e8 /library/cpp/linear_regression/linear_model.h
parent3196904c9f5bf7aff7374eeadcb0671589581f61 (diff)
Restoring authorship annotation for <[email protected]>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/linear_regression/linear_model.h')
-rw-r--r--library/cpp/linear_regression/linear_model.h24
1 files changed, 12 insertions, 12 deletions
diff --git a/library/cpp/linear_regression/linear_model.h b/library/cpp/linear_regression/linear_model.h
index c04f61c4a0d..8bb050cff79 100644
--- a/library/cpp/linear_regression/linear_model.h
+++ b/library/cpp/linear_regression/linear_model.h
@@ -1,32 +1,32 @@
-#pragma once
-
-#include <util/generic/algorithm.h>
-#include <util/generic/vector.h>
-
-#include <util/ysaveload.h>
-
+#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;
}
@@ -39,4 +39,4 @@ public:
double Prediction(const TVector<T>& features) const {
return InnerProduct(Coefficients, features, Intercept);
}
-};
+};